Machine Learning Research Engineer (MLRE) - GPUs
Achira· Machine Learning
Hybrid (San Francisco Office), Hybrid (Remote), Hybrid (New York Office)$164.6k - $259k + equity + bonusFull-timePosted 11mo ago
About the role
Machine Learning Research Engineer focused on high-performance GPU implementations for molecular machine learning, collaborating with scientists and optimizing PyTorch/JAX code, with opportunities to interact with NVIDIA tooling and multi-site teams.
Responsibilities
- Take existing PyTorch and JAX, profile and optimize it without compromising model accuracy, reproducibility, and robustness with respect to scientific objectives.
- Develop with frameworks like CUDA, Triton, Warp, etc. to accelerate performance critical code sections.
- Liaise with NVIDIA to represent our needs and implement their tooling in our environment.
- Work day-to-day with scientists to identify areas of greatest impact, including travel to our SF and NY working groups to collaborate.
Qualifications
- Engineer with at least two years professional experience in GPU optimization.
- Deep understanding of GPU programming fundamentals.
- Solid track record of observable artifacts (e.g., GitHub) showing optimization work.
- Experience collaborating on software projects across multi-person teams.
Skills
Required:GPU optimizationGPU programmingPyTorchJAXCUDA
Nice to have:NVIDIA tooling
Benefits
- Offers Equity
- Offers Bonus
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) - Workflows/Systems5mo agoHybrid (San Francisco Office), Hybrid (New York Office) · Machine Learning