San Francisco, United States of America On-site Full-time USD 350,000 – 475,000 / year

Thinking Machines Lab is hiring a Software Engineer, Data Infrastructure

Responsibilities

  • Design, build, and operate scalable, fault-tolerant infrastructure for LLM Research: distributed compute, data orchestration, and storage across modalities.
  • Develop high-throughput systems for data ingestion, processing, and transformation — including training data catalogs, deduplication, quality checks, and search.
  • Build systems for traceability, reproducibility, and robust quality control at every stage of the data lifecycle.
  • Implement and maintain monitoring and alerting to support platform reliability and performance.
  • Collaborate with research teams to unlock new features, improve data quality, and accelerate training cycles.

Requirements

  • Bachelor’s degree or equivalent experience in computer science, engineering, or similar.
  • Proficiency in at least one backend language (we use Python or Rust).
  • Are fluent in distributed compute frameworks such as Apache Spark or Ray.
  • Are deeply familiar with cloud infrastructure, data lake architectures, and batch and streaming pipelines.
  • Comfort operating across the stack and owning projects end-to-end.
  • Thrive in a highly collaborative environment involving many, different cross-functional partners and subject matter experts.
  • A bias for action with a mindset to take initiative to work across different stacks and different teams where you spot the opportunity to make sure something ships.

Nice to Have

  • Have hands-on experience with Kafka, dbt, Terraform, and Airflow.
  • Have experience building a web crawler.
  • Have extensive experience understanding and scaling deduplication, data mining, and search.
  • Have strong knowledge of file formats and storage systems (e.g., Parquet, Delta Lake, etc.) and how they impact performance and scalability.
  • Are proactive about documentation, testing, and empowering your teammates with good tooling.

Benefits

  • generous health, dental, and vision benefits
  • unlimited PTO
  • paid parental leave
  • relocation support as needed

Work Arrangement

On-site — San Francisco, California

Additional Information

  • This is an 'evergreen role' that we keep open on an on-going basis to express interest.
  • We continuously review applications and reach out to applicants as new opportunities open.
  • You may reapply if you gain more experience, but please avoid applying more than once every 6 months.
  • You are welcome to apply to project or team specific roles in addition to this evergreen role.

Visa sponsorship available

Required Skills
Apache KafkaTerraformAirflow
About company
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.

While AI capabilities have advanced dramatically, key gaps remain. The scientific community’s understanding of frontier AI systems lags behind rapidly advancing capabilities. Knowledge of how these systems are trained is concentrated within the top research labs, limiting both the public discourse on AI and people’s abilities to use AI effectively. And, despite their potential, these systems remain difficult for people to customize to their specific needs and values. To bridge the gaps, we’re building Thinking Machines Lab to make AI systems more widely understood, customizable and generally capable.

We are scientists, engineers, and builders who’ve created some of the most widely used AI products, including ChatGPT and Character.ai, open-weights models like Mistral, as well as popular open source projects like PyTorch, OpenAI Gym, Fairseq, and Segment Anything.

All jobs at Thinking Machines Lab Visit website
Job Details
Category data
Posted 4 hours ago