Updated 18h ago

Machine Learning Intern at Helical · AI Job Signal

As a Machine Learning Engineer intern at Helical, you'll build, optimize, and scale real-world applications of bio foundation models.

You'll work closely with researchers and product engineers to productionize model training, inference, and deployment workflows. You'll also help push the limits of foundation models by prototyping new methods, contributing to our core ML infrastructure, and translating research into fast, iterative code.

This is a deeply technical role with high ownership - ideal for engineers who want to operate at the bleeding edge of AI infrastructure, model development, and system design.

What You'll Do

Build and maintain scalable training/inference pipelines for various biological foundation models architectures (e.g. Transformers, SSMs).
Optimize model performance, latency, and throughput across environments.
Apply the latest post-training research to current architectures and data setups.
Design modular, reusable ML components for internal and open-source use.
Collaborate with researchers to turn notebooks into production-grade systems.
Own ML infrastructure components (data loading, distributed compute, experiment tracking, etc.).
Drive experimentation around fine-tuning, transfer learning, and evaluation at scale.

Requirements

Essentials

MSc or PhD in Machine Learning, Computer Science, Applied Math, or similar.
Strong Python programming skills, with deep knowledge of PyTorch, JAX, or TensorFlow.
Hands-on experience building and scaling ML pipelines in real-world settings.
Comfort with MLOps tools and practices (e.g. Weights & Biases, Ray, Docker, etc.).
Experience with modern ML architectures - Transformers, Diffusion Models, SSMs, etc.
High agency, fast iteration speed, and comfort with ambiguity in early-stage environments.

Bonus Points