
AI/ML Engineer
Kaizen Labs3 days ago
Base Salary
$240k - $280k/yr
Responsibilities
- Fine-tune, adapt, and distill open-weight models for software-engineering-related tasks.
- Own the model experimentation loop, including dataset curation, training pipelines, architecture experiments, and technique evaluation.
- Build evaluations on real software engineering tasks to establish baselines, expose failure modes, and measure genuine improvements.
- Collaborate with platform engineers to identify model limitations and ship capabilities into production.
- Experiment with supervised fine-tuning, synthetic data, preference optimization, and reinforcement learning.
- Deliver customer-focused applications that reduce reliance on expensive frontier-model inference.
- Establish datasets, evaluations, and research practices for future in-house model training.
Requirements
- Demonstrated hands-on experience training, fine-tuning, or adapting language models using model weights rather than only model APIs or prompts.
- Strong software engineering fundamentals, including building reliable training pipelines, debugging complex systems, and writing reusable code.
- Strong Python and PyTorch skills, with working knowledge of transformer architectures, tokenization, and language-model training.
- A demonstrated model-engineering project, such as a fine-tuned model, original training approach, or meaningful open-source ML contribution.
- Preferred qualifications include experience with code generation, program synthesis, repository understanding, synthetic data, distillation, quantization, preference optimization, reinforcement learning, model development, applied ML research, or reproducing significant research results.
Benefits
- Medical, dental, and vision coverage is fully paid for employees and dependents, with Oxford/United, Guardian PPO, and Beam plans.
- Includes fully paid life insurance, FSA and Dependent Care FSA, One Medical membership, fertility and family-building support, and a 401(k) with a 2% company match.
- Provides 16 weeks of paid parental leave for birthing parents and 10 weeks for non-birthing parents.
- Offers unlimited PTO with a two-week minimum, federal holidays, a company-wide winter break, and company offsites.
- Hybrid New York City work arrangement with in-office work Monday through Thursday; candidates must live in or commute to New York City.
- Provides home-office or desk setup stipends, commuter benefits, expensed office lunches, late-work dinner and Uber rides, and a company laptop.
- Includes gym or fitness reimbursement, utility, home cleaning, professional development, recreation, and pet-care stipends.