Early Career Machine Learning Engineer, Applied AI

About Brain Co.

About the role

What you will do

• Design and deploy LLMs to automate complex manual processes.
• Build scalable data pipelines; optimize models for performance and accuracy; prepare models for production; monitor and maintain deployments.
• Contribute to projects involving energy production systems, government workflows, and public healthcare outcomes.
• Work directly with government officials and collaborate with founders, AI researchers, and software engineers to deliver AI solutions.
• Take on varied work across software building, product management, sales, and interpersonal collaboration.
• Follow AI and ML developments, participate in code reviews, share knowledge, and promote high-quality engineering.

Candidate profile

• 0–2 years of industry experience in applied machine learning or related AI work.
• Bachelor's, master's, or PhD in Computer Science, Machine Learning, Data Science, or a related field is listed as a candidate qualification.
• Hands-on Generative AI application experience, such as agents, reasoning workflows, or RAG, and understanding of LLM architecture and operation.
• Personal model implementation experience in PyTorch, JAX, or TensorFlow.
• Strong grounding in data structures, algorithms, and software-engineering principles.
• Problem-solving and analytical skills, collaboration, adaptability to competing priorities, and willingness to own problems end to end are sought.

Benefits

• Benefits listed include medical, dental, and vision coverage, paid parental leave, 401(k), daily lunches, commuter benefits, and unlimited PTO.
• The posting also mentions competitive salary and meaningful equity, without giving a salary amount or pay cadence.

Work environment

• The role works with cross-functional colleagues and may involve direct engagement with government officials in multiple countries.
• No explicit remote, hybrid, or onsite policy is stated.