Senior AI/ML Engineer – Post-Training Platform
Hybrid (Pasadena or San Francisco)Full-timePosted 3mo ago
About the role
Senior-level AI/ML engineer focusing on post-training platform infrastructure for Bonsai models, including fine-tuning, reinforcement learning, evaluation, orchestration, and model lifecycle management across cloud and partner-hosted environments.
Responsibilities
- Building scalable systems for fine-tuning, reinforcement learning, evaluation, and post-training workflows for Bonsai models
- Developing infrastructure for data ingestion, dataset preparation, orchestration, artifact storage, logging, telemetry, and cost tracking
- Supporting post-training techniques including LoRA, full fine-tuning, PPO, GRPO, DPO, and related optimization workflows
- Building reliable multi-tenant infrastructure with strong isolation, access control, observability, and production reliability
- Developing systems for model evaluation, benchmarking, experiment tracking, and model lifecycle management
- Collaborating with model, infrastructure, and product teams to improve training efficiency, usability, and deployment readiness
- Translating advances in post-training workflows and AI infrastructure into robust, production-ready platforms
Qualifications
- 5–8+ years of experience in machine learning systems, distributed systems, infrastructure engineering, or related fields
- Strong programming skills in Python and experience building production-quality AI/ML systems
- Hands-on experience building infrastructure for fine-tuning, reinforcement learning, or large-scale AI workflows
- Solid understanding of distributed systems, orchestration, and modern AI/ML pipelines
- Experience deploying AI/ML systems in cloud or production infrastructure environments
- Familiarity with observability, monitoring, and debugging production systems
- Proven ability to mentor and collaborate effectively with other engineers
Skills
Required:PythonDistributed systemsPost-training techniques (LoRA, PPO, DPO, GRPO)
Nice to have:Model evaluation and lifecycle management
Last updated Oct 22, 2026. Interested? Apply on the company site.
About PrismML
Concentrating intelligence
PrismML — Concentrating intelligence
Bonsai 27B Bonsai 8B Bonsai 4B Bonsai 1.7B Bonsai Image
More open roles at PrismML
- Senior AI/ML Engineer – Forward Deployment2mo agoHybrid (Pasadena; San Francisco) · PrismML Careers / Engineering
- Senior AI/ML Engineer – Kernel Optimization3mo agoHybrid (Pasadena;San Francisco), Remote · Engineering
- Staff AI/ML Engineer – Large-Scale Systems3mo agoHybrid (Pasadena or San Francisco)