# Research Engineer, Asta

**Company:** [Ai2](https://hotfix.jobs/companies/ai2)
**Location:** Seattle, WA
**Role:** ML Engineering
**Salary:** $119k – $178k/yr
**Experience:** 2+ years
**Skills:** Python, PyTorch, JAX, TensorFlow, Machine Learning, Natural Language Processing, Reinforcement Learning, Docker, GCP, AWS, Agentic Frameworks
**Posted:** 2026-05-13

> Research Engineer builds ML infrastructure and agentic systems to accelerate scientific discovery in biology, neuroscience, and more. Requires 2+ years experience with Python, PyTorch/JAX/TF, cloud resources, and deep learning for LLM/agent research at Ai2.

## Job Description

## Responsibilities
- Building infrastructure to facilitate the next generation of LLM and agentic research.
- Creating AI tools to facilitate scientific discovery in domains such as biology, cancer research, neuroscience, social science, etc.
- Designing, building, and training machine learning or language models for agentic workflows.
- Bridging the gap between cutting-edge research and a widely adopted product.
- Bringing software engineering best practices to a research environment.
- Supporting and collaborating with an open-source community.
- Releasing contributions as open source software, model releases, additions to Ai2’s public API and open research datasets, and technical reports.

## Requirements
- Bachelor’s degree in CS/EE/Data Science/Applied Mathematics/Statistics/ML/NLP, or related field, or equivalent experience.
- **2+ years** building agentic infrastructure that handles tools, skills, and other artifacts.
- **2+ years** building infrastructure for data preprocessing/transformation and machine learning model training, evaluation, inference, and deployment.
- Knowledge of modern deep learning, natural language processing, and reinforcement learning techniques.
- Strong software engineering skills, particularly building performant systems and debugging.
- Experience with **Python** and **PyTorch/Jax/TensorFlow**, agentic frameworks (e.g., MCP).
- Familiarity with cloud compute resources (**GCP**, **AWS**, **Modal**) and containerization (**Docker**).

## Nice-to-Haves
- Advanced degree in CS/EE/Data Science/Applied Mathematics/Statistics/ML/NLP or related fields.
- Contributions to open-source ML or research libraries (e.g., spaCy, AllenNLP, transformers, langchain).
- Experience operating at scale in production settings.
- Experience in HPC settings.

## Compensation
**Base salary range: $118,800 - $178,200**, plus generous bonus plans.

## Similar roles

- [Quantitative Researcher](https://hotfix.jobs/jobs/4e070f2c-1ae6-4443-9347-82d26724abdc) - Trexquant - New York, NY - $120k – $180k/yr
- [Machine Learning Engineer](https://hotfix.jobs/jobs/04313290-8e0d-49b1-9c43-81cfbb0d9789) - Mariana Minerals - Ann Arbor, MI - $120k – $160k/yr
- [Founding AI Engineer](https://hotfix.jobs/jobs/09a63ffc-a750-417f-9010-aad2ebfde73f) - Bild AI - San Francisco, CA - $120k – $180k/yr
- [Deep Learning Compiler Engineer (New Grad)](https://hotfix.jobs/jobs/58350561-91a6-45e2-9ca2-1fb9043b6e2f) - Quadric - Burlingame, CA - $120k – $160k/yr
- [Data Scientist, New Grad](https://hotfix.jobs/jobs/ad9b60a4-1a32-4c53-a549-288c4210e0ed) - SentiLink - Remote - $120k – $220k/yr

**Apply:** https://hotfix.jobs/jobs/50766069-cf57-4073-b1bd-a9b997b92021
**Canonical:** https://hotfix.jobs/jobs/50766069-cf57-4073-b1bd-a9b997b92021