Machine Learning Engineer — Distillation
Featherless AI4 months ago
Remote, WorldwideMid Level / Senior
Responsibilities
- Design and implement knowledge distillation pipelines.
- Distill large foundation models into smaller, faster models for inference.
- Run and analyze large-scale training experiments to evaluate tradeoffs.
- Collaborate with research to translate distillation ideas into production code.
- Optimize training and inference performance.
- Contribute to internal tooling and experiment tracking.
Requirements
- Strong background in machine learning or deep learning.
- Hands-on experience with model distillation.
- Solid understanding of training dynamics and optimization.
- Experience with PyTorch or JAX and modern ML tooling.
- Comfort running experiments on multi-GPU or distributed setups.
- Ability to reason about model quality vs. performance tradeoffs.
Benefits
- Work on core model quality and cost efficiency.
- High ownership and direct impact on product and roadmap.
- Small, senior team with strong research and engineering culture.
- Competitive compensation and meaningful equity.
- Remote-friendly, async-first environment.
Tech Stack
PyTorch
Categories
AI & MLData Science