Machine Learning Engineer, Platform

The role

You will build core machine learning capabilities for the Atlas platform. This includes developing document extraction, foundation models, model routing, and unified evaluation systems used across all company products.

What you will do

• Turn pod needs into platform capabilities
• Own capabilities end-to-end from development to production behavior
• Turn LLMs, RL fine-tuning, and agentic systems into capabilities
• Serve project pods and domain experts as customers
• Engineer for production reality including accuracy, latency, cost, and reliability
• Turn discoveries from one pod into shared learnings for every pod

Required qualifications

• Understand machine learning philosophy including loss functions, generalization, and distribution shift
• Experience with LLMs and agentic systems including prompting, fine-tuning, tool use, and reasoning
• Ability to compose segmentation models, VLMs, and rule engines into systems
• Platform instinct to identify general capabilities within specific requests
• Ability to build things that have never existed
• Comfortable when problems, data, and success definitions must be invented

Benefits

• Competitive salary
• Meaningful equity
• Country-specific benefits

About the company

Brain Co. builds AI-native operating systems for large, regulated institutions. Their proprietary platform, Atlas, powers agents that automate workflows across industries like government, insurance, health, and financial services.