ML Infrastructure Engineer

Echo; San Francisco (CA); Hybrid; USD 180,000 - 230,000; Full time

Benefits offered by this job

Stock options

401(k) matching

Job summary

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

The role focuses on developing production-grade ML infrastructure to support rapid experimentation with neural signals

Qualifications

• Bachelor’s degree in CS/EE or related technical field.
• 5+ years in software engineering or large-scale data infrastructure.
• Extensive proficiency in Python and PyTorch.
• Experience with distributed-training frameworks and large models.
• Ability to partner with research teams to productionize workflows.

Responsibilities

• Create flexible and performant ML cloud infrastructure for massive-scale modeling and analytics.
• Support diverse model exploration, hyperparameter optimization, pretraining, fine-tuning, and evaluation.
• Design scalable distributed training pipelines with model sharding and cross-GPU communication.
• Build and maintain robust ML platforms across the model lifecycle.
• Make architecture decisions balancing performance, cost, reliability, and scalability.
• Design and optimize data platforms and pipelines for scalable data access.
• Enable data ingestion from internal and external datasets with varied formats.
• Ensure reproducibility and documentation of design decisions and evaluations.

Skills

Python

PyTorch

Distributed training frameworks

Large-scale data pipelines

Model training pipelines

Cross-functional collaboration

Cloud infrastructure design

Education

Bachelor's degree in CS/EE or related field

Advanced degree (MS/PhD) preferred

Tools

Kubernetes

Docker

CUDA

Go / C++ (optional)

ML tooling (DeepSpeed, Megatron-LM, Ray)

Job description

Echo Neurotechnologies is an exciting new startup in the Brain-Computer Interface (BCI) space, driving innovation through advanced hardware engineering and AI solutions. Our mission is to deliver cutting‑edge technologies that restore autonomy to people living with disabilities and improve their quality of life.

Team Culture

Join a small, dedicated team of knowledgeable and motivated professionals. Our early‑stage environment offers the opportunity to take ownership of broad decisions with significant and long‑lasting impact. We emphasize continuous learning and growth, fostering cross‑functional collaboration where your contributions are vital to our success.

SUMMARY

We are seeking a Senior Machine Learning Infrastructure Engineer to join our team. The person who fills this role will design, build, and scale infrastructure to power massive‑scale data, modeling, and analysis platforms, playing a critical role in shaping a high‑performance, production‑grade ML ecosystem to support rapid experimentation with diverse datasets spanning neural signals, behavior, and more. This person will have significant ownership over the ML R&D platform, working closely with domain experts to architect new cloud infrastructure, data pipelines, and modeling flows. The work will ultimately enable the development of cutting‑edge models for neuroscientific discovery and neural decoding, empowering brain‑computer interface technology to improve the lives of patients living with severe neurological conditions.

PRIMARY ROLE AND RESPONSIBILITIES

• Create flexible and performant ML infrastructure
  • Design and build systems 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 support for features such model sharding, cross‑GPU communication, and real‑time training monitoring
  • Create, operate, and maintain robust ML platforms and services across the model lifecycle
  • Make informed architecture decisions that balance performance, cost, reliability, and scalability
• Build diverse and scalable data platforms
  • Design, build, and optimize massive‑scale databases and data pipelines for scalable, flexible, and reliable data access
  • Explore research‑driven, tailored data solutions using existing and simulated data, comparing performance and efficiency across solutions for typical data‑access patterns
  • Create infrastructure and pipelines for ingesting internal and external datasets with varied shapes, formats, and associated metadata
  • Design and assess custom data formats for efficient storage and slicing of high‑dimensional time‑series data
  • Enable efficient data movement, preprocessing, and artifact management for data lineage and modeling reproducibility
• Meet company standards for delivered solutions
  • Establish best practices for reliability, observability, reproducibility, and operational excellence across the ML ecosystem
  • Make informed and collaborative decisions with domain experts across the software & ML teams
  • Foster visibility and reproducibility within the company by maintaining extensive documentation of design decisions, evaluations of viable alternatives for selected solutions, pipeline assessments, etc.
  • Support ML R&D operations while preparing for eventual incorporation into product pipelines

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

OTHER QUALIFICATIONS

• Advanced degree (MS or PhD) in Computer Science, Electrical Engineering, or a related technical discipline
• Proficiency in C++, Go, CUDA, Rust, and/or Java
• Experience in data engineering and systems ML for time‑series data
• Deep understanding of the fundamentals of distributed systems, including scalability, fault tolerance, monitoring, observability, scheduling, performance tuning, and resource management
• Experience with cloud‑native environments and orchestration (Kubernetes, Docker, etc.)
• Experience scaling foundation‑model training infrastructure or multi‑cluster computing environments

$180,000 - $230,000 USD

What We Offer

An opportunity to work on exciting, cutting‑edge projects to transform patients’ lives in a highly collaborative work environment.

Competitive compensation, including stock options.

401(k) program with matching contributions.

Equal Opportunity Employer

Echo Neurotechnologies is an Equal Opportunity Employer (EOE). We celebrate diversity and are committed to creating an inclusive environment for all employees.