Inception Labs
Member of Technical Staff, ML Product Engineering
San Mateo, United States · On-site · Staff · Full-time · Posted Mar 10, 2026
- Experience
- At least 5 years of experience working on ML projects in PyTorch (or equivalent).
- Education
- BS/MS/PhD in Computer Science, Machine Learning, or a related field (or equivalent experience).
About the job
Train and deploy diffusion large language models, building customer-facing ML products and ensuring reliable production performance at scale.
What you'll do
- Design, develop, and optimize models for production; contribute to deployment and maintenance.
- Partner with customers and product teams to translate requirements into technical solutions and ML features.
- Develop post-training generative AI approaches, including agentic workflows, and work on data pipelines, evaluation, and alignment.
What they ask for
- At least 5 years of experience on ML projects using PyTorch or equivalent.
- Strong familiarity with transformers and core LLM concepts; experience training and fine-tuning LLMs.
- Familiarity with large-scale systems and high-performance computing, including GPU/TPU utilization.
- Experience with Git and Docker; strong communication skills for technical and non-technical stakeholders.
Nice to have
- Expertise in data engineering and synthetic data generation for LLMs; knowledge of MLOps and production deployment.
- Experience with LLM serving frameworks, cloud platforms, or model quantization and optimization.
Tools
PyTorch, transformers, LoRA, KV caching, GPU/TPU, Git, Docker, vLLM
AI focus
Train, fine-tune, post-train, evaluate, align, optimize, and deploy diffusion LLMs, including agentic workflows and synthetic data work.
Model-written summary · the original posting is authoritative