Machine Learning Research Engineer (MLRE) - GPUs
- Location
- San Francisco, CA (Remote)
- Job type
- Full-time
- Level
- Senior
- Experience
- 2+ years
- Listed
- 23 days ago
Skills & highlights for this role
- CUDA
- PyTorch
- GPU optimization
Also useful
- JAX
- Triton
- distributed compute
About the Role
We're looking for a rare individual who thrives at the intersection of cutting-edge deep learning architectures and high-performance computing. You will help shape the future of molecular machine learning by engineering high-efficiency implementations of advanced architectures for foundation simulation models, accelerating simulations to the limits of the hardware while maintaining fidelity to the underlying physics.
While we prefer candidates willing to relocate to San Francisco or New York City, we are willing to consider full-time remote candidates of exceptional talent who are willing to travel frequently to our two office sites. Travel is part of all roles at Achira, both to conferences and corporate on-site activities.
What You’ll Do
- 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.
About You
- 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.
Nice to Have
Even if you hit none of these bonus features, we encourage you to apply!
- Experience working with multi-cloud distributed compute systems.
- Experience working with multi-site distributed company team.
- Experience working with equivariant architectures that operate on 3-D point clouds.
Why Achira
At Achira, we are building a team of world-class scientists, ML researchers, and engineers to work together to move beyond the beaten path in drug discovery. We are actively exploring the next frontier of model architectures for AI x Chemistry: developing world models for the physical microcosm. Our goal is to make biology at the molecular level something that can be learned, predicted, and designed.
At Achira, you’ll operate at the frontier scale of massive compute, massive data, and massive ambition. You’ll own impactful work end-to-end, from ideation to architecture to deployment on distributed infrastructure. We are a well-funded, talent-dense organization that values rigor, speed, execution, and an ownership mindset. We’re looking for new members who share our sense of relentless urgency and are natural collaborators who value team success.
From Achira's careers page, first listed on Rolev.ai on October 17, 2026. Rolev.ai is not the employer.
How Rolev.ai helps with this application
Rolev.ai is a job search and application tool for US job seekers. For this Machine Learning Research Engineer (MLRE) - GPUs role at Achira, it shows how your resume fits, tailors your resume to the posting, and prepares the application for you to review.
See how you fit
Openings from employer career sites are ranked against your whole resume—and searchable inside Claude, ChatGPT, and Cursor via Rolev.ai MCP.
Tailor without inventing
Every change to your resume is checked against your original. New experience goes in only with your yes—zero fabricated claims.
Draft the full kit
Tailored and Full Match resumes, role-specific cover letters, and answers to employer screening questions are prepared in seconds.
1-Click Extension Autofill
The Chrome extension fills the employer's Greenhouse, Workday, Lever, or Ashby form in 1 click. You review it and press Submit.
Similar roles
All Machine Learning Engineer jobs- Machine Learning EngineerLucid Motors · Southfield, MI
- Machine Learning Engineer 4Capital One · McLean, VA
- Member of Technical Staff - Research Engineer, RL EnvironmentsPreference Model · San Francisco, CA
- Machine Learning Research Intern (Summer 2027)Scale AI · New York, NY
- Senior Machine Learning Engineer, Defect Quality & SystemsSec · Taylor, TX
- ML EngineerAstrocade · Palo Alto, CA
More at Achira
All Achira jobsCommon questions
Where is the Machine Learning Research Engineer (MLRE) - GPUs job at Achira located?
Yes, this Machine Learning Research Engineer (MLRE) - GPUs opening at Achira is listed as San Francisco, CA (Remote). Always check Achira's application page for specific state eligibility or relocation policies.
What qualifications are required for Machine Learning Research Engineer (MLRE) - GPUs at Achira?
For the Machine Learning Research Engineer (MLRE) - GPUs role, Achira highlights 2+ years of relevant experience; experience with CUDA, PyTorch, GPU optimization. Rolev.ai can score your existing resume against these requirements and show any skill gaps before you apply.
How do I apply for the Machine Learning Research Engineer (MLRE) - GPUs position at Achira?
This listing links directly to Achira's Ashby application page. You can apply directly on Achira's site, or use Rolev.ai to check your resume fit, generate a zero-hallucination tailored resume and cover letter for this posting, and autofill the Ashby application form in 1 click with the Rolev.ai Chrome extension. You always review the form and press Submit yourself.