A

Achira

Atomistic drug-discovery simulation platform

ML Research Scientist - MLRS, Representation Learning for Molecular AI

AI ResearchConfirmed live in the last 24 hours
Full-time$164.6k - $259k /yr

About the Job

Requirements

  • A drive to apply modern machine learning techniques to solve problems at the frontier of the microscopic world.
  • A willingness to follow the data and take an empirical approach to the bitter lesson.
  • A pragmatic approach to inductive bias, such as physical priors and equivariance, in model building.
  • Experience designing, running, and analyzing machine learning experiments at scale.
  • Experience with three-dimensional geometric deep learning.
  • Demonstrated research impact through conference talks or publications in machine learning venues, open-source contributions, or released models.
  • Strong interdisciplinary communication and presentation skills, including the ability to translate ideas and concepts to colleagues from non-machine-learning backgrounds.
  • Proficiency in Python and modern machine learning frameworks, including PyTorch and JAX.
  • Experience collaborating on research projects across multi-person teams.
  • Desire and comfort working on frontier problems in physical AI to invent the blueprint for how they will be tackled.

Responsibilities

  • Build a robust pre-, mid-, and post-training curriculum that ensures foundation model performance and impact.
  • Create reinforcement learning strategies to help models focus their capacity where it matters most, especially when training data does not cover the domain of applicability.
  • Develop expressive representations of molecular and atomistic structure and dynamics, including equivariant graph neural networks, geometric transformers, and latent encoders that capture physical symmetries and constraints.
  • Prototype, benchmark, and iterate rapidly to transform research ideas into reusable and scalable components across Achira’s ecosystem.
  • Collaborate with physicists and chemists to ensure models are grounded in real physics.
  • Work with research engineers and the infrastructure team to identify where research ideas need support to deliver effective results.

Desired Qualifications

  • Professional post-degree experience as a machine learning researcher in an industry setting.
  • Experience with self-supervised representation learning techniques.
  • Experience with Bayesian deep learning techniques and uncertainty quantification.
  • Experience developing or applying generative models for three-dimensional systems.
  • Experience with equivariant graph neural network architectures, such as NequIP, MACE, SchNet, and PaiNN, or similar.
  • Prior experience working in or with researchers in computational chemistry, biology, or materials science.
  • Experience working with multi-cloud distributed compute systems.
  • Experience working with a multi-site distributed company team.

About the Company

Achira

Atomistic drug-discovery simulation platform

Achira develops atomistic foundation simulation models for drug discovery by integrating geometric deep learning with physics, quantum chemistry, and statistical mechanics to create advanced potentials and generative models. It generates large, accurate synthetic datasets free of experimental artifacts, enabling researchers to pursue inverse design by optimizing compounds directly from target properties. The approach combines physics-based simulations with data-driven learning, setting it apart from firms that rely solely on limited data or standalone ML models. Achira aims to shorten drug discovery timelines and lowers costs by providing reliable inverse-design solutions to pharmaceutical and biotech partners through partnerships, licensing, and collaborations.

Company size
11-50
Company stage
Seed
Total funding
$33M
Headquarters
New York City, New York
Founded
2024

Recently Posted Jobs

Explore more opportunities at Achira

See more jobs