AI/ML Engineer

About the role

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 various nation states. This role is your opportunity to make a significant impact by making AI technology both accessible and influential.

In this role

• 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.

Company Momentum

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

What You'll Do

• Design and deploy advanced LLM models to tackle real-world problems, particularly in automating complex, manual processes in a range of real-world verticals.
• Build scalable data pipelines, optimize models for performance and accuracy, and prepare them for production.
• Maintain deployed models to ensure they continue delivering value across various governments worldwide.
• 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.
• 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.

Requirements

• 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.
• Work collaboratively with cross-functional teams.
• Work in fast-paced environments where priorities or deadlines may compete.
• Be 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