Early Career Machine Learning Engineer, Applied AI SF

Brain Co.

San Francisco, Northern (CA, KY)

Hybrid

USD 140,000 - 190,000

Full time

14 days+

Application generator

Don’t send a generic resume — generate a resume and cover letter tailored to this exact role.

Get past ATS filters

Benefits offered by this job

Competitive salary

Medical, Dental, and Vision (100% 1)

Paid Maternity and Paternity Leave

401(k)

Daily Lunches

Commuter Benefits

Unlimited PTO

Job summary

Brain Co. is hiring a Machine Learning Engineer to deploy state‑of‑the‑art AI in healthcare, government, energy, and more. You will turn research breakthroughs into practical, scalable AI solutions for various nation states, collaborating with ex‑Founders and industry veterans.

You’ll design and deploy LLMs, build scalable data pipelines, and ensure models deliver measurable impact across sectors. The role blends software engineering, product sense, and leadership in a fast‑growing company.

Qualifications

0–2 years of industry experience in applied machine learning or related AI work.
BSc/Master’s/PhD in Computer Science, Machine Learning, Data Science, or related field.
Hands‑on experience building GenAI-focused applications and LLM architectures.
Experience with ML frameworks such as PyTorch, Jax or TensorFlow.
Strong foundations in data structures, algorithms and software engineering.

Responsibilities

Design and deploy advanced LLM models to automate complex real‑world processes.
Build scalable data pipelines and optimize models for production deployment.
Monitor deployed models to ensure ongoing value across governments worldwide.
Engage with government officials and senior engineers to deliver AI solutions.
Participate in code reviews and promote high‑quality engineering practices.

Skills

Applied ML

GenAI apps

PyTorch

JAX

TensorFlow

Data structures

Software engineering

Problem solving

Cross-functional teamwork

Education

CS/ML degree

Job description

Our Mission

Rebuild how the world works, to make institutions work better for the people they serve.

About Brain Co.

Brain Co. builds AI-native operating systems for large, regulated institutions. Each system is built for a specific industry, powered by agents that push real workflows forward. Underneath it all is Atlas, our proprietary platform that keeps customers in control, secure by design, and never locked into one model.

Why Now

Brain Co. is entering its next phase of production deployments on a national scale with an elite team built from Palantir, Google, Meta, and Nvidia, and a growing footprint across government, insurance, health, and financial services.

Joining now means shaping both the company and a new category of applied AI. Every project here ships to production and is expected to create measurable customer value and impact.

It’s the kind of work you’ll still be proud of in ten years from now.

About The Role

As a Machine Learning Engineer at Brain Co., you will play a crucial role in deploying state-of-the 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:

Have 0-2 years of industry experience in applied machine learning or related AI work.
Hold a BSc/Master’s/PhD degree in Computer Science, Machine Learning, Data Science, or a related field.
Have hands‑on experience building GenAI-focused applications (e.g., agents, reasoning workflows, or RAG) and a solid understanding of how large language models are architected and operated.
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.