Thinking Machines Lab

Software Development thinkingmachines.ai

Website

X (Twitter)

Browse 2 Thinking Machines Lab Megatron jobs for Aug 2026 in Software Development paying $350K to $475K/year. Work as a Research Engineer, Infrastructure, Numerics and Research Engineer, Infrastructure, Training Systems.

Open Jobs

Salary Range

$350K – $475K

Thinking Machines Lab Megatron Jobs

Airflow Bash Cadence Distributed Systems JAX Kafka Kubernetes Linux Megatron NCCL OpenTelemetry Prototyping RLHF React Reinforcement Learning Research Engineer Rust Software Engineer Spark Technical Documentation TensorFlow Triton TypeScript Workday vLLM

Cities

San Francisco 25 New York 1

Countries

United States 78

Filter by type

Full Time

TML

Research Engineer, Infrastructure, Numerics

Thinking Machines Lab / San Francisco, United States / Hybrid

PyTorch JAX Distributed Systems

$350K – $475K 13D

TML

Research Engineer, Infrastructure, Training Systems

Thinking Machines Lab / San Francisco, United States / Hybrid

PyTorch JAX Megatron

$350K – $475K 13D