Software Engineer, Research Tools
Thinking Machines Lab2 months ago
San Francisco, CA, USA or New York, NY, USAMid Level
H1B sponsor
Base Salary
$300k - $350k/yr
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 with researchers to identify bottlenecks, gather feedback, set priorities, and unlock new capabilities.
- Build reproducible, traceable, observable, and quality-controlled systems for research experiments and model training runs.
Requirements
- Bachelor’s degree or equivalent practical experience in computer science, engineering, machine learning, or a related field.
- At least two years of post-graduate 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, with Python and Rust used primarily, plus React and TypeScript frontend experience.
- 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 problem discovery through deployment and operation.
- Experience collaborating with cross-functional partners and subject-matter experts.
- Preferred qualifications include researcher tooling, developer productivity, polished user-facing applications, startup or small-team experience, ML research infrastructure, and close collaboration with researchers.
Benefits
- Visa sponsorship is offered, with the company committed to working through the visa process for suitable candidates although success is not guaranteed for every candidate or role.
- Generous health, dental, and vision benefits.
- Unlimited paid time off and paid parental leave.
- Relocation support as needed.
- The role is based in San Francisco, California or New York, New York.
Tech Stack
Categories
About Thinking Machines Lab
Thinking Machines Lab develops AI and generative AI software and conducts applied research to help organizations make data-driven decisions. The company builds products and data science solutions for enterprise use cases, pairing foundational models with practical tooling and services across industries. It is privately held and headquartered in San Francisco.