Software Engineer, Research Tools
Build full-stack research infrastructure used to manage training runs, evaluations, experiments, and model visualizations. The role partners closely with researchers and requires strong software engineering fundamentals, cross-stack ownership, and at least two years of post-graduate engineering experience.
About the job
Responsibilities
- Design, build, and maintain research infrastructure, including evaluation frameworks, training systems, experiment-tracking platforms, and visualization tools.
- Work across backend systems, data pipelines, and user-facing applications to deliver tools end to end.
- Partner directly with researchers to identify bottlenecks and unlock new capabilities.
- Treat research tooling as a product by gathering feedback, setting priorities, and measuring adoption.
- Build systems for reproducibility, traceability, and robust quality control across research experiments and model training runs.
- Implement monitoring and observability for research systems.
Requirements
- Bachelor’s degree or equivalent practical experience in computer science, engineering, machine learning, or a related field.
- Two years of post-graduate work experience as a software engineer or ML engineer, excluding internships.
- Strong software engineering fundamentals and experience building reliable, maintainable systems.
- Proficiency in at least one backend programming language; the team primarily uses Python and Rust.
- Frontend experience with React and TypeScript.
- Experience with databases, data warehouses such as ClickHouse, caching systems such as Redis, and other data infrastructure.
- Ability to work across the stack and own projects from discovery through deployment and operation.
- Experience collaborating with cross-functional partners and subject-matter experts.
Nice-to-haves
- Experience building tools for researchers, improving developer productivity, or creating technical products for technical users.
- Experience building polished, intuitive user-facing applications with strong product judgment and attention to UI/UX detail.
- Experience at a startup or on a small team building technically complex products end to end.
- Experience building or maintaining ML research infrastructure, such as training frameworks, evaluation libraries, or experiment-tracking systems.
- Experience working closely with researchers to understand and solve tooling needs.
Compensation and Benefits
- Expected annual salary: $300,000–$475,000 USD, depending on background, skills, and experience.
- Health, dental, and vision benefits.
- Unlimited paid time off.
- Paid parental leave.
- Relocation support as needed.
- Visa sponsorship.
Skills
Python, Rust, React, TypeScript, ClickHouse, Redis, Databases, Data Pipelines, ML Infrastructure, Monitoring And Observability
Similar jobs
Fullstack Engineering jobs
Thinking Machines Lab
San Francisco, CA
Build an end-to-end evaluation platform for frontier AI models, spanning backend systems, data pipelines, APIs, and user-facing applications. The role requires software engineering experience, model evaluation expertise, and the ability to collaborate closely with researchers.
Garner Health
New York, NY
Build and maintain AI-enabled software systems for a healthcare technology company, collaborating across disciplines and protecting sensitive user data. The role requires at least two years of software engineering experience, particularly with distributed event-driven architectures.
Applied Intuition
Sunnyvale, CA
Builds and deploys software and machine learning components for behavior prediction, decision-making, and environmental interaction systems. Requires a bachelor’s degree and at least one year of relevant experience with Python, Go or C++, scalable data pipelines, deep learning, and cloud infrastructure.
The Voleon Group
Berkeley, CASoftware engineering intern designing high-performance systems for real-time trading, distributed data processing, and machine-learning research infrastructure. The role requires strong computer science fundamentals, significant coding experience, and familiarity with Linux, Git, databases, and cloud infrastructure.
Greptile
San Francisco, CA
Build and deploy full-stack features for an AI code-review product, tackling LLM memory, codebase indexing, and semantic search challenges. The role requires a computer science or equivalent degree, at least one year of software or DevOps engineering experience, and JavaScript/TypeScript expertise.