Early Career AI/ML Engineer

Position Overview

Brain Co. is an Applied AI startup founded by Elad Gil and Jared Kushner and backed by many of Silicon Valley's leading builders – including Patrick Collison (CEO of Stripe), Andrej Karpathy (Cofounder of OpenAI), Mike Krieger (CPO of Anthropic), Kevin Weil (CPO of OpenAI), and Aravind Srinivas (CEO of Perplexity). We are building an AI platform and applications for the world's most important institutions, delivering impact on real-world problems. Our progress includes automated construction permitting for a sovereign government, optimized supply chains for a leading global energy company, streamlined hospital patient care across national health systems, a successful $30M Series A raise, and the assembly of a team of 40+ AI experts from Tesla, Google DeepMind, NVIDIA, and Databricks. At Brain Co., your work will be deployed in the real world, not stuck in research. We move fast, with more demand than we can serve, and are looking for exceptional people to take ownership from day one.

As an AI/ML 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 nation states.

Key Responsibilities

• Innovate and Deploy: Design and deploy advanced LLM models to tackle real-world problems, particularly by automating complex, manual processes across various 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 continued value delivery.
• Make a Difference: Engage in projects such as optimizing advanced energy production systems, modernizing government workflows, or improving patient outcomes in public healthcare systems, directly impacting society.
• Engage with Leaders: Interact directly with government officials while working alongside experienced ex-founders, AI researchers, and software engineers to understand complex business challenges and deliver AI-powered solutions.
• Learn and Lead: Stay current with the latest in machine learning and AI, participate in code reviews, share knowledge, and exemplify high-quality engineering practices.

Required Qualifications

• Have 0-2 years of industry experience in applied machine learning or related AI work.
• Hold a BSc, Master’s, or PhD degree in Computer Science, Machine Learning, Data Science, or a related field.
• Possess hands-on experience building GenAI-focused applications (e.g., agents, reasoning workflows, or RAG) and a solid understanding of large language model architectures and operations.
• Have implemented models using ML frameworks such as PyTorch, Jax, or TensorFlow.
• Possess a strong foundation in data structures, algorithms, and sound software engineering principles.
• Exhibit excellent problem-solving and analytical skills with a proactive approach to challenges.
• Demonstrate the ability to work collaboratively with cross-functional teams in fast-paced environments.
• Be eager to own problems end-to-end and willing to acquire any necessary knowledge to get the job done.

Benefits & Perks

• Salary: Competitive salary
• Medical, Dental, and Vision (100% Coverage)
• Paid Maternity and Paternity Leave
• 401(k)
• Daily Lunches
• Commuter Benefits
• Unlimited PTO
• 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 plus meaningful equity in a high-growth company
• Thrive in a culture built on speed, curiosity, and impact