wagey.ggwagey.gg
26,614  jobs26,614  jobs
Browse Tech JobsCompaniesFeaturesPricingFAQs
Log InGet Started Free
Jobs(26,614)/Machine Learning Engineer Role(345)/Achira (2) - Machine Learning Research Engineer (MLRE)
Achira

Achira - Machine Learning Research Engineer (MLRE)

Remote - San Francisco, California, United States$165k - $259k+ Equity3w ago
RemoteNAArtificial IntelligenceMachine Learning EngineerResearch EngineerJAXCUDATritonWarpObservable

Requirements

• 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.

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. • 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.

Benefits

• Independent ContributorBased on qualifications and experience $164,638 – $259,000 • Offers Equity • Offers Bonus • 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. • 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.

Apply in one click

Upload My Resume

Drop here or click to browse · Tap to choose · PDF, DOCX, DOC, RTF, TXT

Apply in One Click
Apply in One Click

Similar roles

AnthropicAnthropic - Research Engineer, Machine Learning (RL Velocity)6mo ago
·Remote-Friendly (Travel-Required) | San Francisco, CA | New York City, NY - Hybrid·$500k - $500k/year + Equity
In OfficeNABankingArtificial IntelligenceResearch EngineerMachine Learning EngineerJAX
AchiraAchira - Machine Learning Research Engineer (MLRE)1w ago
·San Francisco, California, United States - Hybrid·$165k - $259k/year + Equity
In OfficeNAArtificial IntelligenceNonprofitMachine Learning EngineerResearch EngineerDocumentationGitHubPyTorchJAXObservable
relationrxrelationrx - Senior Machine Learning Research Engineer2w ago
·London, United Kingdom
In OfficeEMEASeniorArtificial IntelligenceOil & GasResearch EngineerMachine Learning EngineerPythonCUDATriton
AnthropicAnthropic - Research Engineer, Machine Learning (RL Velocity)6mo ago
·London, UK - Hybrid·£370k/year/year + Equity
In OfficeEMEABankingArtificial IntelligenceResearch EngineerMachine Learning EngineerJAX
HigharcHigharc - Applied Research Engineer3w ago
·Remote - New York, United States·Equity
RemoteNAMidCloud ComputingArtificial IntelligenceResearch EngineerMachine Learning EngineerDocumentationTraining DevelopmentPythonKerasFastAPI
DeepgramDeepgram - Research Engineer, Machine Learning Systems7mo ago
·Remote - California, United States·$150k - $250k/year
RemoteNAInternArtificial IntelligenceResearch EngineerMachine Learning EngineerKubernetesDockerPrefectMLflow
ElicitElicit - Machine Learning Engineer8mo ago
·Oakland, CA, United States·$385k+/year + Equity
In OfficeNAArtificial IntelligenceMachine Learning EngineerResearch EngineerDocumentationPython
10a Labs10a Labs - Machine Learning Engineer4d ago
·Remote - USA *·$130k - $200k/year
RemoteNAMidArtificial IntelligenceMachine Learning EngineerJAXPythonPyTorch
Applied IntuitionApplied Intuition - Senior Machine Learning Engineer3w ago
·Remote - USA·$190k - $260k/year + Equity
RemoteNASeniorArtificial IntelligenceRoboticsMachine Learning EngineerC++CUDAPython

Browse more by category

Show 345 moreMachine Learning EngineerShow 116 moreResearch EngineerShow 93 moreJAXShow 71 moreCUDAShow 46 moreTritonShow 11 moreWarpShow 97 moreObservable
Privacy·Terms··Contact·FAQ·Wagey on X