Research Infrastructure Engineer - AI Tools & Pipelines

Thinking Machines Lab Inc.; San Francisco, Northern (CA, KY); Hybrid; USD 350,000 - 475,000; Full time

Benefits offered by this job

Health benefits

Dental benefits

Vision benefits

Unlimited PTO

Paid parental leave

Relocation support

Job summary

Thinking Machines Lab Inc. is seeking engineers to build the libraries and tools that accelerate AI research.

You will own internal infrastructure, including evaluation libraries, RL training libraries, and experiment tracking platforms, and design scalable systems that boost research velocity. You will work directly with researchers to remove bottlenecks, ensure reproducibility, and deliver reliable tooling.

Qualifications

• Bachelor's degree or equivalent in CS, engineering or ML.
• Strong software engineering fundamentals applicable to reliable systems.
• Backend proficiency in Python or Rust and end-to-end ownership.

Responsibilities

• Design, build, and operate research infrastructure including evaluation frameworks and RL training systems.
• Develop high-throughput, scalable pipelines for distributed evaluation and multimodal assessment.
• Build reproducible, observable systems with quality controls across experiments.
• Partner with researchers to identify bottlenecks and drive adoption of tooling.
• Collaborate with infra, data, and product teams to integrate tools across the stack.

Skills

Software engineering fundamentals

Backend development (Python or Rust)

Collaborative work

Education

Bachelor's degree or equivalent in CS/Engineering/ML

Job description

Thinking Machines Lab Inc. is seeking engineers to build the libraries and tools that accelerate AI research.

You will own internal infrastructure, including evaluation libraries, RL training libraries, and experiment tracking platforms, and design scalable systems that boost research velocity. You will work directly with researchers to remove bottlenecks, ensure reproducibility, and deliver reliable tooling.