Pay
$166K–$240K is typical
The middle half of what similar US postings state. This one doesn’t list pay.
From 1,279 postings in the last year. Stated ranges, not salaries paid.
Machine Learning Research Engineer (MLRE) - GPUs
San Francisco Office · Remote · New York Office · San Francisco, California, United States
Open 5 months, first seen by us 155 days ago- Pay
- Pay not listedSimilar roles state $166K–$240K
- Work setting
Hybrid — work setting source
Location type: Hybrid
From the employer’s posting- Employment
Full-time — employment source
Employment type: Full-time
From the employer’s posting- Team
Machine Learning — team source
Department: Machine Learning
From the employer’s posting
What you’ll work on
Full postingWe're looking for a rare individual who thrives at the intersection of cutting-edge deep learning architectures and high-performance computing.
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.
Profile and optimize existing PyTorch and JAX code without hurting model accuracy
Write CUDA, Triton, or Warp code to speed up performance-critical sections
Work with NVIDIA on tooling and represent Achira's needs to them
Work day-to-day with scientists to find the highest-impact areas
From the employer’s posting
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.
Take existing PyTorch and JAX, profile and optimize it without compromising model accuracy
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
What you’ll bring
All qualificationsCore experience
- At least two years of professional GPU optimization experience
- Deep understanding of GPU programming fundamentals
- Public track record of optimization work, such as on GitHub
- Experience collaborating on software projects in multi-person teams
Qualification wording
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.
Posting record
Report an errorOpen 5 monthsWe first saw this posting 155 days ago and it is still up. Roles open this long are often slow to fill or kept open to collect applications.
- Status in our records
- Live
- First seen by us
- Jun 2, 2026155 days ago
- Recorded sightings
- 69
- Last seen by us
- Nov 5, 2026
- Employer says posted
- Apr 29, 2026
These dates show when we found the posting. Check the employer’s website to confirm it is still accepting applications.
How this job compares
Time since we first saw it
Day 155
Up longer than most. Similar jobs usually come down between day 6 and day 69.
From 4,423 similar postings that came down. Coming down is not always a hire.
Achira is hiring
5 open jobs
- No new data jobs since July
- Its jobs tend to stay up a long time
See how this role fits your experience
Add your resume and we’ll mark which of these tools you already use, and point out anything in the posting that could rule you out.
- PyTorch
Where the posting names them
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.
Your next step
Complete your application on jobs.ashbyhq.com. The employer’s form will show what is required.
Applying through Ashby
- No account needed. “Autofill from resume” at the top of the form fills the key fields from your file.
Already applied? Track this application
About the role
We file this job under ML Engineer, by the work the posting describes.
ML engineers put machine learning models into products. They write the code that trains, ships, serves and monitors models so a recommendation or prediction works for real users.
Postings ask for strong Python and software engineering, a framework such as PyTorch, cloud services, and more and more often experience with large language models.
A data scientist often works out which model to build; an ML engineer makes it run reliably in production.
US government figures for computer and information research scientists
- Median yearly pay, May 2025
- $140,300
- Expected change, 2025 to 2035
- 21.8% more jobsagainst 3.5% for all jobs
The government has no code for ML engineers, so these figures are for the closest group, computer and information research scientists.
Source: US Bureau of Labor Statistics, Occupational Employment and Wage Statistics (May 2025) and Employment Projections (2025 to 2035). Occupation code 15-1221.
Job description
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.
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.
Source details
Source notes
Source excerptsSelected passages from the saved posting. Check the full description for conditions and exceptions.
- Pay
No pay amount identified in the saved description.
- Location & working pattern
San Francisco, California, United States
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
- Work authorization
No clear work-authorization passage found. Eligibility is unconfirmed.