Senior Software Engineer, Computer Vision (Knowledge Distillation)
Is this job for you?
Build my CV Build your CV and see how well you match this role — and every other one.
The role
This role focuses on designing and implementing highly efficient AI systems for generative tasks around diffusion models and vision, with emphasis on training stability and scalable deployment.
You will advance algorithm acceleration through knowledge distillation, architecture redesign (dynamic MoE routing, sparse attention), and low-bit quantization.
Responsibilities include optimizing diffusion sampling, latent optimization, and autoregressive components for faster, cheaper inference.
You will prototype and ship hardware-efficient models and tooling, enabling large-scale multi-GPU, multi-machine deployment.
Strong collaboration with ML research, infrastructure, and product teams is essential to translate GenAI capabilities into products.
Ideal candidates bring a track record in diffusion models, model compression, and efficient architectures, plus excellent communication in fast-paced settings.
You will advance algorithm acceleration through knowledge distillation, architecture redesign (dynamic MoE routing, sparse attention), and low-bit quantization.
Responsibilities include optimizing diffusion sampling, latent optimization, and autoregressive components for faster, cheaper inference.
You will prototype and ship hardware-efficient models and tooling, enabling large-scale multi-GPU, multi-machine deployment.
Strong collaboration with ML research, infrastructure, and product teams is essential to translate GenAI capabilities into products.
Ideal candidates bring a track record in diffusion models, model compression, and efficient architectures, plus excellent communication in fast-paced settings.
See the full job post
Responsibilities, requirements, skills and benefits — create your free account.
Already have an account? Log in
Similar openings
Other roles that could suit you.
? Intern, Machine Learning Engineer ? Senior Data Scientist - Product Analytics (Payments/Ecommerce) ? Data Scientist Graduate (TikTok-Product-Data Science)-2026 Start (PhD) ? [HackerRank] Software Engineer Graduate (On-Device AI) - 2025 Start (BS/MS) ? Senior Director, Risk Data Science ? Data Scientist Graduate (TikTok-Product-Data Science)-2026 Start (BS/MS)
Remote workno
CitySan Jose, United States