Member of Technical Staff, Machine Learning Engineer
About The Position
Edison Scientific builds and deploys AI scientist agents to accelerate science and the development of new medicines. The company is an ambitious team of scientists and engineers from leading institutions across biology, physics, chemistry, and AI. As a Member of Technical Staff, Machine Learning Engineer, you will build models and agents to accelerate scientific discovery. The role combines cutting-edge research with practical engineering, bridging advanced machine learning concepts and robust, reliable software used by scientists. You will work closely with AI researchers, chemists, and biologists in a multidisciplinary environment focused on innovation and discovery.
Requirements
- 5–10+ years of a strong track record in applied ML research and applying ML methods to real-world problems.
- Experience across the ML lifecycle, including data pipelines and provenance, model training, model deployment, and validation in production systems.
- Fluency in PyTorch, JAX, or an equivalent framework.
- Demonstrated experience with experimentation in academic or industry settings.
- Strong programming expertise and the ability to adapt to technical challenges across the data, ML, and LLM software stack.
Nice To Haves
- PhD in Machine Learning, Computer Science, or another quantitative field.
- Familiarity with leveraging and managing distributed computing resources.
- Background architecting complex distributed systems.
Responsibilities
- Interpret qualitative challenges in building AI agents for science as well-formulated optimizable problems.
- Build environments in which to train and deploy AI agents that solve scientific tasks.
- Work with scientists to formulate and scale training data pipelines, ensuring observability and reproducibility.
- Lead training of large-scale LLM-based systems, including building internal infrastructure to improve the efficiency of experimentation and production training runs.
- Build efficient and flexible inference infrastructure to support complex sampling algorithms and custom architectures.
- Develop and extend the experimentation platform for internal tools and projects.
- Collaborate with AI researchers, chemists, and biologists.
Benefits
- Competitive salary and equity.
- Full healthcare coverage; the company pays 100% of premiums for employees and dependents.
- Yearly new parent stipend and fertility coverage through Carrot.
- Mental health support through Rula, an in-network therapist and psychiatrist network.
- 12 weeks of paid parental leave for maternity, paternity, and adoption.
- Yearly employer-funded pet care stipend.
- Commuter benefits for transit and parking using pre-tax dollars.
- 401(k) company matching.
- $300 quarterly health and wellness benefit.
- Lunch on office days and dinner when working late.
- Regular team off-sites and company events.
Stand Out From the Crowd
Upload your resume and get instant feedback on how well it matches this job.
What This Job Offers
Job Type
Full-time
Career Level
Senior
Education Level
No Education Listed
