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.
About the job
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
Python, PyTorch, JAX, TensorFlow, Machine Learning, Natural Language Processing, Reinforcement Learning, Docker, GCP, AWS, Agentic Frameworks
Similar jobs
ML Engineering jobs
Simple AI
San Francisco, CA
Build and deploy production voice AI agents for customers, creating demos, debugging edge cases, improving performance, and translating feedback into product improvements. The role combines hands-on engineering, customer engagement, and pre- and post-sales delivery.
MongoDB
United States
Build and maintain tooling, evaluation systems, quality gates, and infrastructure for MongoDB's agent skills and AI platform. Requires 2+ years building production software, developer tools, CLIs, test infrastructure, or CI/CD, with strong fundamentals in API design, testing, and reasoning about nondeterministic AI systems.
DataVisor
Mountain View, CA
Builds high-scale data pipelines, distributed systems, and AI agent workflows using LLMs for fraud intelligence platform. Requires 2+ years software engineering, Python proficiency, big data tools, AWS/K8s, and ML foundations.
PathAI
Boston, MADevelop and deploy machine learning models for biomedical research and AI product development, collaborating with scientific, engineering, and product teams. Requires a master's degree with 2–4 years of experience or a PhD with 0–2 years, plus strong Python and ML development skills.
Benchling
San Francisco, CA
Build datasets, evaluations, and scalable data systems that improve frontier AI models on challenging biological and scientific tasks. The role partners with scientists and AI labs and requires at least two years of experience applying biology and AI, plus hands-on LLM experience.