ML Engineer - Scaling
Helical.AI Alle Jobs anzeigen
- Luxemburg
- Unbefristet
- Vollzeit
- Build and maintain scalable training/inference pipelines for foundation models (e.g. Transformers, SSMs).
- Optimize model performance, latency, and throughput across environments.
- Design modular, reusable ML components for internal and open-source use.
- Collaborate with researchers to scale notebooks into production-grade systems.
- Own ML infrastructure components (data loading, distributed compute, experiment tracking, etc.).
- 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
- Contributions to open-source ML libraries or tooling.
- Experience with distributed training, model compression, or serving at scale.
- Scaling AI Systems For Large Post-Training Runs.
- Knowledge of how to integrate ML systems into user-facing applications or APIs.
- Interest in the biology/pharma space (not required, but you’ll pick it up fast here!).