Machine Learning Engineer, Applied AI

Latitude; San Francisco (CA); On-site; USD 180,000 - 260,000; Full time

Benefits offered by this job

Competitive salary

Medical, Dental, Vision (100% Coverage

Paid Maternity and Paternity Leave

401(k)

Daily Lunches

Commuter Benefits

Unlimited PTO

Job summary

Brain Co. is building frontier AI applications to modernize essential services in governments, healthcare, and critical industries. As a Machine Learning Engineer, you will deploy state-of-the-art models to automate real-world problems and translate research breakthroughs into practical, scalable solutions.

You will work alongside ex-founders, AI researchers, and software engineers to deliver impactful AI-powered outcomes, influencing how institutions leverage AI at scale.

Qualifications

• Hold a degree in Computer Science, Machine Learning, Data Science, or related field.
• Demonstrated experience building GenAI-focused applications with latest technologies (Agents, reasoning models, RAG).
• Strong familiarity with large language model architectures and operation.
• Hands-on experience implementing models in PyTorch, Jax or TensorFlow.
• Solid foundation in data structures, algorithms, and software engineering principles.

Responsibilities

• Innovate and Deploy: design and deploy advanced LLM models for real-world problems across healthcare, government, and energy sectors.
• Optimize and Scale: build scalable data pipelines, optimize models for performance and accuracy, and prepare them for production.
• Monitor and maintain deployed models to ensure continued value across governments worldwide.
• Engage with leaders: collaborate with government officials and senior researchers to understand challenges and deliver AI-powered solutions.
• Learn and Lead: keep up with ML/AI advancements, participate in code reviews, and promote high-quality engineering practices.

Skills

GenAI focus

LLM architectures

Scaling ML systems

Cross-functional collaboration

Problem solving

Education

BSc/Master's/PhD in CS/ML/DS

Tools

PyTorch

Jax

TensorFlow

Job description

About Brain Co.

Brain Co. is an applied AI startup co-founded by Jared Kushner and Elad Gil, and backed by leading Silicon Valley builders including Patrick Collison and Andrej Karpathy.

We are building AI applications for the world's most important institutions, delivering impact on real-world problems across governments, healthcare systems, and critical industries.

Our Progress So Far

• Automated construction permitting for a sovereign government → 80% faster, unlocking $375M+ in value
• Optimized supply chains for a leading global energy company → 30% lower cost, 99% reliability, preventing $100M+ in losses
• Streamlined hospital patient care across national health systems → 40% better outcomes, 80% less admin work

Company Momentum

• Raised a $55M Series A from leading investors
• Built a team of 70+ AI experts from Tesla, Google DeepMind, NVIDIA, and Databricks

At Brain Co., we focus on applying frontier AI to real institutional challenges, working alongside governments, healthcare systems, and critical industries to modernize how essential services operate.

We are looking for leaders who want to help bring new technology into institutions that impact millions of people.

About The Role

As a Machine Learning Engineer at Brain Co., you will play a crucial role in deploying state-of-the-art models to automate various real world problems in sectors such as healthcare, government and energy. Part of the role will involve turning research breakthroughs into practical solutions for various nation states. This role is your opportunity to make a significant impact by making AI technology both accessible and influential.

In This Role, You Will

• Innovate and Deploy: Design and deploy advanced LLM models to tackle real-world problems, particularly in automating complex, manual processes in a range of real-world verticals.
• Optimize and Scale: Build scalable data pipelines, optimize models for performance and accuracy, and prepare them for production. Monitor and maintain deployed models to ensure they continue delivering value across various governments worldwide.
• Make a Difference: Engage in projects including but not limited to optimizing the world's most advanced energy production systems, modernizing core government workflows, or improving patient outcomes in advanced public healthcare systems. Your work will directly impact how AI benefits individuals, businesses, and society at large.
• Engage with Leaders: interact directly with government officials in various countries and apply the first of its kind AI solutions while working alongside experienced ex. Founders, AI researchers, and software engineers to understand complex business challenges and deliver AI-powered solutions. Join a dynamic team where ideas are exchanged freely and creativity flourishes. You will be able to wear many hats: software building, product management, sales, interpersonal skills.
• Learn and Lead: Keep abreast of the latest developments in machine learning and AI. Participate in code reviews, share knowledge, and set an example with high-quality engineering practices.

You Might Thrive In This Role If You

• Hold a BSc/Master's/PhD degree in Computer Science, Machine Learning, Data Science, or a related field.
• Have impactful industry experience building GenAI-focused applications with the latest technologies, including but not limited to Agents, reasoning models and RAG.
• Have at least a high level familiarity with the architecture and operation of large language models.
• Have personally implemented models in common ML frameworks such as PyTorch, Jax or TensorFlow.
• Possess a strong foundation in data structures, algorithms, and software engineering principles.
• Exhibit excellent problem-solving and analytical skills, with a proactive approach to challenges.
• Can work collaboratively with cross-functional teams.
• Thrive in fast-paced environments where priorities or deadlines may compete.
• Are eager to own problems end-to-end and willing to acquire any necessary knowledge to get the job done.

Benefits

• Competitive salary
• Medical, Dental, and Vision (100% Coverage)
• Paid Maternity and Paternity Leave
• 401(k)
• Daily Lunches
• Commuter Benefits
• Unlimited PTO

Why Join Us

• Ship quickly, iterate constantly and see your work deployed at global scale
• Collaborate with industry veterans from Tesla, DeepMind, Databricks, and more
• Accelerate your career with ownership based on impact, not tenure
• Earn competitive compensation + meaningful equity in a high-growth company
• Thrive in a culture built on speed, curiosity, and impact

If you want to see your work deployed at scale with real impact, Brain Co. is the place to build.