Senior/Principal Machine Learning Engineer

About the Company

Edison Scientific builds and commercializes AI agents for science. We are assembling a team of top researchers and engineers across AI and biology to build an AI scientist to accelerate scientific discovery.

Responsibilities

• Interpret qualitative challenges in building AI agents for science as well-formulated optimizable problems
• Build appropriate environments in which to train and deploy AI agents that solve scientific tasks
• Work with scientists to formulate training data pipelines and scale them, ensuring observability and reproducibility
• Lead training of large-scale LLM-based systems, including building internal infrastructure to improve experimentation and production training runs
• Build efficient and flexible inference infrastructure, supporting complex sampling algorithms and custom architectures
• Develop and extend the experimentation platform for internal tools and projects
• Collaborate closely with a multidisciplinary team of AI researchers, chemists, and biologists

Requirements

• 5-10+ years of experience in applied ML research and real-world ML applications
• Experience across the ML lifecycle: data pipelines, model training, deployment, and production validation
• Fluency in PyTorch, Jax, or equivalent frameworks
• Demonstrated experience with experimentation in academic or industry settings
• Strong programming expertise across the data, ML, and LLM software stack

Preferred Qualifications

• PhD in Machine Learning, Computer Science, or a quantitative field
• Familiarity with leveraging and managing distributed computing resources
• Background architecting complex distributed systems

Benefits

• Competitive salary and equity
• 100% healthcare premium coverage for employees and dependents
• Yearly new parent stipend and fertility coverage
• 12 weeks of paid parental leave
• Mental health support through Rula
• Pet care support stipend
• 401(k) company matching
• Quarterly health and wellness benefit
• Daily lunch and late-night dinner provided