Member of Technical Staff, ML Product Engineering

• Design, develop, and optimize our models for production use cases.
• Partner with customers to understand their requirements and translate them into technical solutions.
• Implement innovative approaches for post-training generative AI models, including agentic workflows.
• Work on data preprocessing pipelines, model evaluation, and alignment to enterprise use cases.
• Contribute to the deployment and maintenance of models in production environments.
• Collaborate with product teams to design and implement customer-facing ML features.

• BS/MS/PhD in Computer Science, Machine Learning, or a related field (or equivalent experience).
• At least 5 years of experience working on ML projects in PyTorch (or equivalent), preferably in a research lab or engineering role.
• Excellent familiarity with transformers and core LLM concepts (autoregressive pretraining, instruction tuning, in-context learning, LoRA, KV caching).
• Experience training LLMs, including fine-tuning.
• Familiarity with large-scale systems and high-performance computing, including GPU/TPU utilization.
• Experience with version control (Git) and containerization (Docker).
• Excellent communication skills with the ability to explain technical concepts to non-technical stakeholders.

• Expertise in data engineering and synthetic data generation for LLMs.
• Knowledge of MLOps and production-level deployment workflows.
• Experience with LLM serving frameworks like vLLM, SGLang, or TensorRT.
• Experience with cloud platforms (AWS, GCP, Azure).
• Experience with model quantization and optimization techniques.

Compensation

• Work with World-Class Talent: Collaborate with the inventors of diffusion models and leading AI researchers
• Shape Foundational Technology: Your decisions will influence how the next generation of AI products are built and used
• Immediate Impact: Join at the ground floor where your contributions directly shape product direction and company trajectory

• Competitive salary and equity in a rapidly growing startup
• Flexible vacation and paid time off (PTO)
• Health, dental, and vision insurance
• 401k match
• Catered meals (breakfast, lunch, & dinner)
• Commuter subsidies
• A collaborative and inclusive culture