Technical Lead, Machine Learning
BjakInsurance Comparison company
United StatesLead
BASS Investment
Data & AI
About the role
TL;DR
Lead ML system execution, translating research into production-grade, scalable ML systems.
- •A1 is building a proactive smart assistant to bring intelligence to everyday applications, aiming to reduce task completion time significantly.
- •The product focuses on high reliability for long-running workflows, persistent context, and real-world task completion.
- •Key Responsibilities Own end-to-end ML system execution, including data pipelines, training, evaluation, inference, and deployment.
- •Fine-tune and adapt models using state-of-the-art methods.
- •Architect and operate scalable inference systems, balancing latency, cost, and reliability.
- •Implement evaluation pipelines covering performance, robustness, safety, and bias.
- •Own production deployment, including GPU optimization and latency reduction.
- •Requirements Experience shipping real ML systems used by people.
- •Comfort working with large models and understanding their failure modes.
- •Strong production-grade code and care for system correctness.
- •Self-directed, pragmatic, and ownership-driven.
- •Clear communication and collaboration skills in small, high-trust teams.
Required skills
PythonPyTorchJAXLLMs
Domain expertise
productivityai
Tech stack
PythonPyTorchJAX