Skip to content
Ai2Ai2Seattle, WA

Research Engineer, Asta

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.

119k – 178k/yr
On-site2+ YOEML Engineering

About the role

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.

Skills

PythonPyTorchJAXTensorFlowMachine LearningNatural Language ProcessingReinforcement LearningDockerGCPAWSAgentic Frameworks

Similar roles

ML Engineering jobs
Trexquant

Quantitative Researcher

TrexquantNew York, NY

Early-career Quantitative Researcher developing ML models and trading signals to predict asset movements and build market-neutral portfolios. Requires STEM degree, Python fluency, and passion for machine learning; embedded in Alpha, Data Science, or Strategy teams.

120k – 180k/yr
On-siteEntry levelML Engineering
Mariana Minerals

Machine Learning Engineer

Mariana MineralsAnn Arbor, MI +2

Build and deploy reinforcement learning models to autonomously control mineral refining facilities, optimizing recovery rates, energy use, and uptime in real operating plants.

120k – 160k/yr
On-siteEntry levelML Engineering
Bild AI

Founding AI Engineer

Bild AISan Francisco, CA

Founding AI Engineer building and shipping production-grade computer vision, LLM, and multimodal models for real-world construction documents at an early-stage startup.

120k – 180k/yr
On-siteEntry levelML Engineering
Quadric

Deep Learning Compiler Engineer (New Grad)

QuadricBurlingame, CA

Develops and optimizes deep learning compiler passes for Quadric's GPNPU, lowering ONNX models through Relay IR to efficient C++ code. Requires new grad-level proficiency in Python, C++, and compiler concepts like IR transformations and debugging.

120k – 160k/yr
On-siteEntry levelML Engineering
SentiLink

Data Scientist, New Grad

SentiLinkAustin, TX +5

Build and deploy ML models for fraud detection and financial risk at SentiLink. New grad role for PhD/early-career researchers handling full ML lifecycle with Python, AWS; requires strong ML/stats foundation and production coding skills.

120k – 220k/yr
RemoteEntry levelML Engineering