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
120 days ago
Salary
0
Seniority
Senior
Job Description
Technical Lead, Machine Learning
BJAK
• 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
Related Guides
Related Job Pages
More Machine Learning Engineer Jobs
Lead Instructor – Machine Learning Data Associate
Correlation OneCorrelation One is a technology company that is on a mission “to create equal access to data-driven jobs of tomorrow.” As an employer, the company is known for its empowering,
• Deliver live, virtual instruction to large groups of learners (100 to 8,000+) • Conduct large synchronous online lectures on technical content • Prepare and lead virtual classroom sessions • Collaborate with operations personnel to ensure smooth program delivery • Assist in lesson design, development, and improvement • Interact professionally with learners and staff • Maintain a high level of courtesy and respect • Prepare diligently for lectures to ensure high-quality content • Adjust lesson pace and presentation to meet diverse learner needs • Provide thoughtful answers and assistance to learners throughout the lesson
• Build platforms for driving intelligent decisions, interacting with machine learning models, and powering interactions such as conversational interfaces • Collaborate with data scientists, data engineers, data analysts, software engineers, IT specialists, and stakeholders to expand effective use of AI applications • Research and prove out new approaches and algorithms through prototypes • Ensure high-quality system delivery and continually improve workflows and methods
• Develop, deploy, and maintain robust AI and machine learning pipelines for internal and client-driven projects. • Deploy, manage, and scale AI models, including pre-trained models (e.g., LLMs) and custom ML models, into production environments. • Work with data scientists to implement model prototypes into scalable, production-ready AI systems. • Optimize and tune model performance, latency, and cost-efficiency on cloud platforms. • Integrate AI/ML solutions with major cloud platforms (AWS, GCP, Azure) and utilize containerization technologies (Docker, Kubernetes) for consistent deployment. • Apply standard MLOps practices, including continuous integration/continuous delivery (CI/CD), model versioning, monitoring, and maintenance systems. • Collaborate with software engineering teams to ensure seamless integration of AI capabilities into applications and user workflows. • Stay up to date with advancements in AI/ML technologies, Generative AI, and MLOps deployment strategies.
• Design, develop, and maintain high-performance AI and machine learning pipelines for complex internal and client-driven projects. • Deploy, manage, and scale diverse AI models, including pre-trained models (e.g., LLMs, vision models) and custom ML models, into production environments. • Collaborate with data scientists and researchers to operationalize models, transforming prototypes into resilient, production-grade AI systems that meet stringent business requirements. • Optimize model performance, latency, and cost-efficiency across various cloud and edge deployment targets. • Integrate AI/ML solutions with major cloud platforms (AWS, GCP, Azure) and leverage containerization technologies (Docker, Kubernetes) for consistent deployment. • Implement comprehensive MLOps practices, including automated testing, continuous integration/continuous delivery (CI/CD) for models, model versioning, monitoring, and proactive maintenance systems. • Work closely with software engineering teams to ensure seamless and ethical integration of AI capabilities into larger applications and user workflows. • Stay up to date with the latest advancements in AI/ML research, Generative AI, data engineering, and MLOps deployment strategies.



