Featherless AI
Machine Learning Engineer — Training Optimization
Rol remoto de Machine Learning Engineer con fit claro de ubicación del candidato.
Publicado25 jul 2026
Países elegiblesGlobal
Señal de seniorityMiddle
Modelo de trabajoRemoto
Ubicaciones aceptadas para candidatos
Global
Resumen del rol
Machine Learning Engineer — Training Optimization
Requisitos y responsabilidades
Contenido del rol extraído en secciones para revisar más rápido.
What You’ll Do
- Optimize large-scale model training pipelines (throughput, convergence, stability, and cost)
- Improve distributed training strategies (data, model, and pipeline parallelism)
- Tune optimizers, schedulers, batch sizing, and precision (bf16 / fp16 / fp8)
- Reduce training time and compute cost via profiling, bottleneck analysis, and systems-level improvements
- Collaborate with researchers on architecture-aware training strategies
- Build and maintain robust training infrastructure (checkpointing, fault tolerance, reproducibility)
- Evaluate and integrate new training techniques (e.g. gradient checkpointing, ZeRO, FSDP, custom kernels)
- Own training performance metrics and continuously push them forward
What We’re Looking For
- Strong experience training large neural networks (LLMs or similarly large models)
- Hands-on experience with training optimization (not just model usage)
- Solid understanding of:Backpropagation, optimization algorithms, and training dynamicsDistributed systems for ML training
- Backpropagation, optimization algorithms, and training dynamics
- Distributed systems for ML training
- Experience with PyTorch (required)
- Comfort working close to hardware (GPUs, memory, networking constraints)
- Ability to move fluidly between research ideas and production-ready code
Solid understanding of:
- Backpropagation, optimization algorithms, and training dynamics
- Distributed systems for ML training
Nice to haves
- Experience with large-scale distributed training (multi-node, multi-GPU)
- Familiarity with DeepSpeed, FSDP, Megatron, or custom training stacks
- Experience optimizing training on AMD or NVIDIA GPUs
- Contributions to open-source ML infrastructure or research codebases
- Exposure to non-Transformer architectures (RNNs, hybrid models, etc.)
Why Join Us
- Real ownership at Series-A stage — your work shapes the company’s trajectory
- Work on cutting-edge models and training systems at scale
- Small, highly technical team with fast feedback loops
- Strong emphasis on engineering quality and research rigor
- Competitive compensation + meaningful equity
Roles similares
Mantén una lista de respaldo.
Stack
Usa estas tags para comparar roles remotos similares.
Elegibilidad de ubicación
Este rol aparece como abierto para candidatos en cualquier país.
Tu perfilPaís no definidoInicia sesión para comparar tu país con este rol.
Global
Flujo de contratación
WithMira muestra el rol y luego envía candidatos a la aplicación de la empresa.
1Revisa fit del rol, stack y elegibilidad de ubicación en WithMira.
2Abre la página de aplicación de la empresa desde el link rastreado.
3Guarda el rol o suscríbete a oportunidades similares antes de salir.