Machine Learning Research Engineer (MLRE) - Research

Achira; San Francisco (CA); Sur place; USD 120 000 - 160 000; Plein temps

Résumé du poste

Achira is seeking a talented individual for a hybrid role focused on applied machine learning research and software engineering in San Francisco. You will design and run experiments, engineer metrics, and maintain libraries for large-scale research initiatives. Ideal candidates have a strong software engineering background and solid knowledge of PyTorch and JAX. This role encourages collaboration, with opportunities for travel and teamwork. Join us to contribute to pushing the boundaries of molecular simulation.

Qualifications

• Experience in building reproducible pipelines for research.
• Track record of observable artifacts in ML or scientific computing.
• Comfortable with HPC or large-scale compute environments.
• Sufficient scientific depth to engage with research questions.

Responsabilités

• Design and run experiments for foundation model development.
• Engineer metrics for rapid model iteration.
• Build and maintain libraries for training and experimentation.

Connaissances

Software engineering fundamentals

PyTorch

JAX

Large-scale compute environments

Description du poste

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 applied machine learning research and rigorous software engineering. You will advance the state of the art in foundation simulation models by implementing and experimenting with internal and literature-sourced ideas, participating with research teams to scale our ML systems, train and evaluate models, and engineer scientific prototypes into production.

While we prefer candidates willing to work from our San Francisco office, highly skilled candidates may be considered for working from New York City with travel to San Francisco as needed. Both locations are offered as hybrid roles, spending at least some of your time working from the office in collaboration with coworkers. Travel is part of all roles at Achira, both to conferences and corporate on-site activities

What You’ll Do

• Design and run experiments to test out hypotheses on the path to foundation model development.
• Engineer meaningful evals and metrics which enable rapid model iteration.
• Design, build and maintain scalable, reproducible libraries for training, experimentation evaluation, and simulation, in service of large-scale research initiatives.
• Implement model architectures both from the literature and developed in collaboration with our in-house researchers that push the boundaries of molecular simulation.
• Enable agent-driven research and workflows and maintain guardrails on agentic tooling.
• Help prepare manuscripts, software artifacts, and datasets for public release.

About You

• Strong software engineering fundamentals, with experience not just building one-off scripts but reproducible pipelines for research, writing necessary documentation, and observing coding best-practices.
• Track record of observable artifacts (e.g., GitHub, papers) showing work in ML or scientific computing libraries.
• Solid working knowledge of PyTorch and JAX and the modern ML research stack.
• Comfortable with HPC or large-scale compute environments, and used to thinking on the scale of hundreds or thousands (or even more!) fits running at once.
• Sufficient scientific depth to engage with the research questions, whether developed through prior industry experience or during a PhD.

Nice to Have

Even if you hit none of these bonus features, we encourage you to apply!

• Experience with equivariant architectures, geometric deep learning, or GNNs (NequIP, MACE, SchNet, PaiNN, or similar).
• Familiarity with generative modeling: diffusion models, flow matching, score-based methods.
• Regular involvement in open-source ML or scientific computing libraries.
• Experience building agent-driven research, active learning, and data curation pipelines.