← All roles

Software Engineer, Research Acceleration

Thinking Machines Lab· Research Infrastructure (ML Infra)
Apply on company site
Hybrid (San Francisco, CA), Hybrid (New York, NY)$350k - $475kPosted 8mo ago

About the role

Software Engineer role focused on building libraries and tools to accelerate AI research, including evaluation libraries, RL training libraries, experiment tracking platforms, and research infrastructure. Collaborative role partnering with researchers to identify bottlenecks and deliver reliable, scalable tooling that accelerates research velocity.

Responsibilities

  • Design, build, and operate research infrastructure including evaluation frameworks, RL training systems, experiment tracking platforms, visualization tools, and shared utilities.
  • Develop high-throughput, scalable pipelines for distributed evaluation, reward modeling, and multimodal assessment.
  • Build systems for reproducibility, traceability, and robust quality control across research experiments and model training runs. Implement monitoring and observability.
  • Partner directly with researchers to identify bottlenecks and unlock new capabilities. Own research tooling like a product manager, proactively seeking feedback and tracking adoption.
  • Collaborate with infrastructure, data, and product teams to integrate tools across the technical stack.

Qualifications

  • Bachelor's degree or equivalent experience in computer science, engineering, machine learning, or similar.
  • Strong software engineering fundamentals with a track record of building reliable, maintainable systems.
  • Proficiency in at least one backend language (Python or Rust).
  • Comfort operating across the stack and owning projects end-to-end.
  • Experience in highly collaborative environments involving many different cross-functional partners and subject matter experts.

Skills

Required:Backend developmentDistributed systemsObservability and monitoringCross-functional collaboration

Benefits

  • generous health, dental, and vision benefits
  • unlimited PTO
  • paid parental leave
  • relocation support as needed
Last updated Jul 8, 2026. Interested? Apply on the company site.
Thinking Machines Lab logo

About Thinking Machines Lab

Connectionism: Research Blog by Thinking Machines Lab

Thinking Machines Lab is an artificial intelligence research and product company. We’re building a future where everyone has access to the knowledge and tools to make AI work for their unique needs and goals.

AIartificial intelligencebessemer-ai-infrastructure-2026deep learningliquid-ai-recruiting-targetmachine learningproductreinforcement-learning-platforms

More open roles at Thinking Machines Lab