B

Bridge Neurotech

Machine Learning Software Engineer Intern

InternshipUpdated on 10/28/2026
No salary listed
PhD
Redwood City, CA, USA
In Person
AI & Machine LearningApplied Machine Learning

About the job

Requirements
  • Currently enrolled in a PhD program in machine learning, computer science, electrical engineering, computational neuroscience, physics, or a related field.
  • Available full time for six months.
  • Strong Python and hands-on PyTorch development, or an equivalent framework such as JAX or TensorFlow.
  • Experience taking a model from a paper or prototype to something that trains reliably on real data.
  • Comfort with large numerical datasets and memory-mapped arrays.
  • Solid grounding in first principles, such as linear algebra and signal processing.
  • Good software engineering practices, including version control, testing, and code review.
  • Self-directed and able to execute on a goal rather than needing a step-by-step guide.
Responsibilities
  • Design, implement, train, and evaluate models for one part of the pipeline: functional ultrasound reconstruction, signal preprocessing, or downstream decoding.
  • Run experiments by forming and testing hypotheses about why a model is not learning, designing ablations, and tuning architectures, losses, and training schedules.
  • Extend models to new data regimes, such as 2D to 3D volumetric data, new probes, or new subjects, and build the preprocessing needed to train on them.
  • Contribute to training and evaluation infrastructure, including experiment tracking, metrics, and regression checks that assess whether a model is better.
  • Debug numerical and performance problems, such as NaNs under mixed precision, silent shape bugs, and GPU memory ceilings.
  • Work directly with imaging scientists and neuroscientists to understand data sources and artifacts.
  • Reproduce results from relevant literature and present findings to the team.
Desired Qualifications
  • Experience with neural operators (FNO), implicit neural representations, or physics-informed machine learning architectures.
  • Exposure to neural decoding, brain-computer interfaces, or neuroimaging data.
  • Experience with mixed-precision or multi-GPU training.
  • Publications at machine learning or neuroscience venues, or contributions to open-source machine learning tooling.

About the company

Company Size

N/A

Company Stage

N/A

Total Funding

N/A

Headquarters

N/A

Founded

N/A