Machine Learning Engineer — Distillation

Featherless AI; Poland; On-site; PLN 255,754 - 341,005; Full time

Benefits offered by this job

Competitive compensation

Meaningful equity

Remote-friendly environment

Job summary

A technology company is seeking a Machine Learning Engineer focused on model distillation in Poland. This hands-on role requires designing distillation pipelines and running large-scale experiments to optimize models for production. Ideal candidates will have a strong background in machine learning, experience with PyTorch, and the ability to reason about performance trade-offs. The company offers competitive compensation and a remote-friendly work environment, ideal for those looking to make a significant impact in a small, agile team.

Qualifications

• Strong background in machine learning or deep learning.
• Hands-on experience with model distillation techniques.
• Solid understanding of training dynamics, loss functions, and optimization.

Responsibilities

• Design and implement knowledge distillation pipelines.
• Run and analyze large-scale training experiments.
• Collaborate with research to translate ideas into production-ready code.

Skills

Machine learning

Deep learning

Model distillation

PyTorch

Distributed setups

Tools

PyTorch

JAX

Job description

About the Role

We’re looking for a Machine Learning Engineer focused on model distillation to help us build smaller, faster, and more efficient models without sacrificing quality. You’ll work at the intersection of research and production—taking cutting‑edge techniques and turning them into systems that scale.

This is a hands‑on role with real ownership: you’ll design distillation pipelines, run large‑scale experiments, and ship models used in production.

What You’ll Do

• Design and implement knowledge distillation pipelines (teacher–student, self‑distillation, multi‑teacher, etc.)

• Distill large foundation models into smaller, faster, and cheaper models for inference

• Run and analyze large‑scale training experiments to evaluate quality, latency, and cost tradeoffs

• Collaborate with research to translate new distillation ideas into production‑ready code

• Optimize training and inference performance (memory, throughput, latency)

• Contribute to internal tooling, evaluation frameworks, and experiment tracking

• (Optional) Contribute back to open‑source models, tooling, or research

What We’re Looking For

• Strong background in machine learning or deep learning

• Hands‑on experience with model distillation (LLMs or other neural networks)

• Solid understanding of training dynamics, loss functions, 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

• Pragmatic mindset: you care about shipping, not just papers

Nice to Have

• Experience distilling LLMs or large sequence models

• Experience with inference optimization (quantization, pruning, kernels, etc.)

• Familiarity with evaluation for language models

• Open‑source contributions or research publications

• Experience in early‑stage or fast‑moving startups

Why Join

• Work on core model quality and cost efficiency—not side projects

• High ownership and direct impact on product and roadmap

• Small, senior team with strong research + engineering culture

• Competitive compensation + meaningful equity

• Remote‑friendly, async‑first environment