BJAK
Bjak is a technology company focused on making financial services easy, fun and more rewarding for everyone
Technical Lead, Machine Learning
Location
United States
Posted
23 days ago
Salary
Not specified
Bachelor DegreeEnglish
Job Description
• 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
Job 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
