ML Research Engineer: Molecular Simulation & Drug Discovery

Achira; San Francisco (CA); Hybrid; USD 120,000 - 160,000; Full time

Job summary

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.

Responsibilities

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

Skills

Software engineering fundamentals

PyTorch

JAX

Large-scale compute environments

Job description

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.