Machine Learning Research Engineer (MLRE) - Workflows/Systems
Achira· Machine Learning
Hybrid (San Francisco Office), Hybrid (New York Office)$164.6k - $259kFull-timePosted 5mo ago
About the role
MLRE role focused on architecture and distributed workflows for molecular ML, enabling scalable experiments from data generation to evaluation in drug discovery workflows at Achira.
Responsibilities
- Build and maintain robust multi-stage asynchronous workflows for running data generation, training, and evaluations for our machine learning stack.
- Rationalize machine learning systems design and software architecture.
- Identify blockers and build solutions that scale to the size of foundation models.
- Operate as the glue between research scientists and the infrastructure team.
Qualifications
- At least two years relevant industry experience.
- Highly fluent in and enthusiastic about PyTorch and JAX.
- Used to thinking in asynchronous primitives.
- Strong views on library design: clean abstractions, minimal surface area, consistency.
- Solid track record of observable artifacts (e.g., GitHub) showing clear, well-documented code.
- ML generalist who knows what scalable, reliable ML systems look like.
Skills
Required:PyTorchJAXAsynchronous programmingSoftware architecture
Last updated Jul 15, 2026. Interested? Apply on the company site.
About Achira
Building foundation simulation models for drug discovery
Achira is building atomistic foundation simulation models to power the future of drug discovery.
drug discoverysimulation modelsmachine learningquantum chemistrybiomolecular applications
More open roles at Achira
- People Operations Manager1mo agoHybrid (San Francisco Office), Hybrid (New York Office) · Operational
- ML Research Scientist (MLRS) - Representation Learning for Molecular AI3mo agoHybrid (San Francisco Office), Hybrid (New York Office) · Machine Learning
- ML Research Scientist (MLRS) - Generative AI3mo agoHybrid (San Francisco Office), Hybrid (New York Office) · Machine Learning
- Machine Learning Research Engineer (MLRE) - Research5mo agoHybrid (San Francisco Office), Hybrid (New York Office) · Machine Learning
- Machine Learning Research Engineer (MLRE) - GPUs11mo agoHybrid (San Francisco Office), Hybrid (Remote) +1 more · Machine Learning