Thinking Machines Lab · 3 hours ago
Software Engineer, Research Acceleration
San Francisco
Full-time
Onsite
Mid Level
$350K/yr - $475K/yrThinking Machines Lab's mission is to empower humanity through advancing collaborative general intelligence. They are seeking engineers to build the libraries and tools that accelerate research, focusing on internal infrastructure and collaborating with researchers to enhance research velocity.
Artificial Intelligence (AI)Foundational AIGenerative AIInformation TechnologyProduct ResearchSoftware
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
Qualification
Required
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 (we use 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
Preferred
Track record building tooling for researchers that achieved high adoption without top down mandates
Experience building or maintaining ML research infrastructure such as training frameworks, evaluation libraries, or experiment tracking systems
Contributions to open-source ML tools or widely-used internal frameworks at research-focused organizations
Record of publications or technical writing on ML systems, infrastructure, or tooling
Background working closely with ML researchers to understand and solve their tooling needs
Familiarity with distributed systems, modern ML frameworks (PyTorch, JAX), and data processing at scale
Experience with research observability tools, distributed compute frameworks (Ray, Spark), or large-scale evaluation pipelines
Benefits
Generous health, dental, and vision benefits
Unlimited PTO
Paid parental leave
Relocation support as needed
Company
Thinking Machines Lab
Thinking Machines Lab is an AI research and product company that aims to increase understanding and customization of AI systems.
Founded in 2025
11-50 employees
H1B Sponsorship
Thinking Machines Lab has a track record of offering H1B sponsorships. Please note that this does not
guarantee sponsorship for this specific role. Below presents additional info for your
reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (9)
Funding
Current Stage
Early StageTotal Funding
$2.01BKey Investors
Andreessen HorowitzMinistry of Economy, Culture and Innovation
2025-06-20Seed· $2B
2025-05-05Grant· $9.98M
Leadership Team
Mira Murati
Co-Founder and Chief Executive Officer
Jonathan Lachman
Founding Head of Operations
Recent News
Latest Technology News | New York Post
2026-01-22
Company data provided by crunchbase