Senior ML Engineer - Foundation Models for Neuroscience

Netholabs Ltd.; San Francisco (CA); On-site; USD 150,000 - 230,000; Full time

Job summary

Netholabs Ltd. is seeking a Senior Machine Learning Engineer to design, train, and iterate on foundation models of brain and behavior across species, using large-scale multimodal neural and behavioral data.

You will own core modeling work, including architecture, distributed training, and representation learning, and you’ll collaborate with the data and research teams to turn findings into production-grade code.

Qualifications

• Trained large deep learning models end-to-end in production or research settings.
• Experience training transformer or other large sequence models with distributed training.
• Strong Python and PyTorch (or JAX) fundamentals with reproducible code.
• Experience working with large, multimodal or time-series data.
• Pragmatic in early-stage environments owning end-to-end work.

Responsibilities

• Design, train, and iterate on large generative models over multimodal data.
• Own training at scale: data loading, distributed training, hyperparameters, evaluation.
• Develop representations that cross species and modalities.
• Turn research findings into production-ready, reproducible code.
• Collaborate with data engineering and research teams on roadmap.

Skills

Deep learning

Transformer models

Python

PyTorch

Distributed training

Time-series data handling

Reproducible model code

JAX

Tools

JAX

Job description

Netholabs Ltd. is seeking a Senior Machine Learning Engineer to design, train, and iterate on foundation models of brain and behavior across species, using large-scale multimodal neural and behavioral data.

You will own core modeling work, including architecture, distributed training, and representation learning, and you’ll collaborate with the data and research teams to turn findings into production-grade code.