Technical Lead, Machine Learning
BJAK
full-time
Posted on:
Location Type: Remote
Location: United States
Visit company websiteExplore more
Job Level
About the role
- Own the execution layer of A1’s intelligence – training pipelines, inference systems, evaluation tooling, and deployment
- Architect and operate scalable inference systems, balancing latency, cost, and reliability
- Build and own end-to-end ML pipelines spanning data, training, evaluation, inference, and deployment
Requirements
- You have built or shipped real ML systems used by people, not just demos
- You are comfortable working with large models and understanding their failure modes
- You write strong, production-grade code and care about system correctness
- You are self-directed, pragmatic, and take full ownership of outcomes
- You communicate clearly and collaborate well in small, high-trust teams
Benefits
- Collaborate closely with application engineering to integrate ML systems cleanly into backend, mobile, and desktop products
- Own production deployment, including GPU optimization, memory efficiency, latency reduction, and scaling policies
- Implement evaluation pipelines covering performance, robustness, safety, and bias, in partnership with research leadership
- Design and maintain data systems for high-quality synthetic and real-world training data
- Fine-tune and adapt models using state-of-the-art methods such as LoRA, QLoRA, SFT, DPO, and distillation
- Build and own end-to-end ML pipelines spanning data, training, evaluation, inference, and deployment
- Work under real production constraints: latency, cost, reliability, and safety
- Make pragmatic trade-offs and ship improvements quickly, learning from real usage
Applicant Tracking System Keywords
Tip: use these terms in your resume and cover letter to boost ATS matches.
Hard Skills & Tools
machine learningML pipelinesinference systemstraining pipelinesevaluation toolingproduction-grade codesystem correctnessscalable systemslarge models
Soft Skills
self-directedpragmaticownershipclear communicationcollaborationteamworktrust