Example Company
Software engineer
- Location
- San Francisco
- Type
- Full time
- Compensation
- $150k-$200k
- Posted
- 2 days ago
Join a team building useful software. Work closely with engineering and design to turn new ideas into products.
Loading job details
Machine Learning Engineering Intern - Neural & Behavioral Data
- Location
- Palo Alto
- Type
- Intern
- Posted
- 4D AGO
About the role
LocationPalo Alto, CA
TeamEngineering
TypeInternship
Apply
We are looking for a Summer 2027 Machine Learning Engineering Intern to build and evaluate models and data pipelines for our neural interface research. This is a high-ownership role. You will take a focused project from data preparation through training, evaluation and usable results, with guidance from our engineers and scientists. Your work may include analyzing neural and other biological signals, estimating movement from video, or connecting behavioral measurements to experimental data. You will work closely with our neuroengineering, electrical engineering and biology teams in our Palo Alto lab.
What You’ll Do
- Build Python pipelines to organize, inspect and preprocess experimental data, including signals, images and video
- Review labels and model predictions, identify data-quality problems, and help improve annotation and screening workflows
- Train and fine-tune models in PyTorch, starting with clear baselines and testing focused improvements
- Create reproducible training, validation and test splits that account for related subjects, sessions and recordings
- Evaluate models with task-appropriate metrics, visual checks and failure-case analysis, not just a single average score
- Test robustness to changes in recording conditions, lighting, backgrounds, noise and missing data
- Compare changes to preprocessing, augmentation and model configuration through controlled experiments
- Turn successful experiments into reusable inference and analysis code, and measure accuracy, runtime and resource use
- Track datasets, code versions, configurations and results so others can repeat your work
- Work with the experimental teams to define useful outputs and explain what a model can and cannot reliably measure
What We’re Looking For
- Must be authorized to work in the U.S.
- Minimum 12-week commitment, full-time, on-site in Palo Alto
- Current Bachelor’s or Master’s student in Computer Science, Electrical Engineering, Bioengineering, Applied Mathematics or a related field
- Hands-on experience building and evaluating an ML model through a research project, substantial course project, personal project or past internship
- Practical Python skills, including working with numerical data and writing code others can understand
- Familiarity with PyTorch or a similar deep-learning framework
- A working understanding of supervised learning, overfitting, data leakage and train/validation/test evaluation
- A step-by-step approach to debugging data and models
- Clear experiment records and an ability to explain results and limitations
- Comfortable working in a fast-paced, ambiguous startup environment
Nice to Have
- Computer vision work such as object detection, segmentation or pose estimation
- Time-series analysis, signal processing or electrophysiology data
- Experience with noisy labels, class imbalance or limited training data
- Git, Linux and reproducible experiment workflows
- GPU training, model profiling or inference optimization
- Dataset annotation tools and data-quality checks
- Coursework or research in neuroscience, neural engineering or biological measurement
What You'll Get
- Hands-on ML work tied to real neural interface experiments
- Mentorship from experienced engineers and scientists
- Ownership of a project from baseline to measured result
- Opportunities to contribute to meaningful projects impacting real patients
Why Join Us
At Axo Neurotech, you’ll be joining a team that believes in bold ideas, scientific rigor, and the power of interdisciplinary collaboration. We’re creating the tools that will define the future of neurotechnology - and we’re just getting started.
Apply for this role See all open roles →Similar roles
Machine Learning Engineer IVAvalara · Remote
Onsite Service Engineer - AI Data CenterDell Technologies · Chennai
Principal Machine Learning Engineer I** Hybrid in Horsham, PA or Remote EST PreferredLexisNexis · Raleigh
Engineering Delivery Manager IIDeloitte
Director, EngineeringDover
Engineering Applications Software Engineer lNASA Jet Propulsion Laboratory