Senior ML Infra Engineer — Scalable AI for Neuroscience

Stealth Neurotechnology Company; San Francisco (CA); On-site; USD 180,000 - 230,000; Full time

Benefits offered by this job

Stock options

Comprehensive benefits

401(k) with matching

Job summary

Echo Neurotechnologies seeks a Senior Machine Learning Infrastructure Engineer to design, build, and scale ML infrastructure for massive-scale data, modeling, and analysis platforms. You will own the ML R&D platform and collaborate with domain experts to architect new cloud infrastructure and data pipelines for neuroscientific discovery.

You will shape a high-performance, production-grade ML ecosystem to support rapid experimentation with neural signals and behavior, enabling brain-computer

Qualifications

• Bachelor's degree in Computer Science, Electrical Engineering, or a related technical discipline.
• 5+ years of industry experience in software engineering, large-scale data infrastructure, or systems ML.
• Extensive proficiency in Python.
• Familiarity with PyTorch.
• Experience designing, building, and maintaining high-throughput data pipelines for large and diverse datasets.
• Experience working with distributed-training frameworks (e.g. FSDP, DeepSpeed, Megatron-LM, Ray, etc.).
• Experience building or optimizing ML training pipelines for transformers or other large neural-network models.
• Demonstrated ability to partner closely with research and modeling teams to productionize workflows.
• Excellent communication and collaboration skills to work effectively on cross-functional and interdisciplinary teams.
• Experience having technical ownership over at least one successfully implemented collaborative project.

Responsibilities

• Create scalable and performant ML infrastructure.
• Design and build ML cloud infrastructure to enable massive-scale modeling and analytics.
• Support diverse model exploration, hyperparameter optimization, pretraining, fine-tuning, and evaluation processes.
• Design and optimize scalable distributed training pipelines with features like model sharding and cross-GPU communication.
• Create, operate, and maintain ML platforms across the model lifecycle.
• Make informed architecture decisions balancing performance, cost, reliability, and scalability.
• Build diverse, scalable data platforms and pipelines for massive datasets.
• Ensure data formats and data movement support reproducibility and lineage.
• Collaborate with domain experts to integrate ML R&D into production-ready pipelines.

Skills

Python

PyTorch

Distributed training

Data pipelines

Cloud ML infra

Collaboration

Education

Bachelor's degree in CS/EE or related

MS or PhD in CS/EE

Tools

FSDP

DeepSpeed

Megatron-LM

Ray

Kubernetes

Docker

Job description

Echo Neurotechnologies seeks a Senior Machine Learning Infrastructure Engineer to design, build, and scale ML infrastructure for massive-scale data, modeling, and analysis platforms. You will own the ML R&D platform and collaborate with domain experts to architect new cloud infrastructure and data pipelines for neuroscientific discovery.

You will shape a high-performance, production-grade ML ecosystem to support rapid experimentation with neural signals and behavior, enabling brain-computer