Skip to content

Thinking Machines Lab

Research, General Agents

Thinking Machines Lab is hiring a Research, General Agents in San Francisco, United States. It pays $400k-$500k a year and Level rates it ; you can apply on Thinking Machines Lab.

Level's read on this job

  • Builds AI ●●●● is how Level rates this role: the job is building AI systems.
  • Thinking Machines Lab mentions visa sponsorship in this posting.
  • Tools and skills the posting cites include PyTorch, TensorFlow and JAX.
  • Level lists 57 other open roles at Thinking Machines Lab.

AI in this role

PyTorchTensorFlowJAX
About Thinking Machines

The mission of Thinking Machines is to build AI that extends human will and judgment. We are training frontier models with Inkling, developing Tinker to let people make models their own, and crafting interfaces that broaden human-AI communication. We believe the future worth building is human, and we're hiring people who want to build it.

About the Role

This role is responsible for advancing the agentic capabilities of our models, with ownership spanning the full development cycle. Our research team is small, and the role carries a corresponding degree of autonomy and responsibility.

Note: This is an "evergreen role" that we keep open on an on-going basis to express interest in this research area. We receive many applications, and there may not always be an immediate role that aligns perfectly with your experience and skills. Still, we encourage you to apply. We continuously review applications and reach out to applicants as new opportunities open. You are welcome to reapply if you get more experience, but please avoid applying more than once every 6 months. You may also find that we put up postings for singular roles for separate, project or team specific needs. In those cases, you're welcome to apply directly in addition to an evergreen role.

What You’ll Do

  • Build a synthetic data framework used across the team, and create new task environments to scale training across agentic capabilities such as tool use, long-horizon tasks, and complex workflows.

  • Address identified gaps through data and recipe work, validating proposed changes through controlled ablations.

  • Enhance usability for agent capabilities end to end, translating internal and external feedback into model improvements.

Skills and Qualifications

Minimum qualifications:

  • Strong engineering skills, ability to contribute code and debug in complex codebases.

  • Ability to design, run, and interpret experiments with scientific rigor and clarity.

  • Proficiency in Python and familiarity with at least one deep learning framework (e.g., PyTorch, TensorFlow, or JAX). Comfortable with debugging distributed training and writing code that scales.

  • Bachelor’s degree or equivalent experience in Computer Science, Machine Learning, Physics, Mathematics, or a related discipline with strong theoretical and empirical grounding.

  • Clarity in communication, an ability to explain complex technical concepts in writing.

Preferred qualifications — we encourage you to apply even if you don’t meet all preferred qualifications, but at least some:

  • Experience building synthetic data pipelines and systems that were adopted by others on your team and remain in use today.

  • Experience owning the end-to-end cycle of identifying gaps in model usability and closing them through custom evaluations and training data.

  • Experience making large-scale agentic RL infrastructure reliable given the long tail of failures that surface at scale.

  • Experience improving the agentic capabilities of a frontier model.

  • PhD in Computer Science, Machine Learning, Physics, Mathematics, or a related discipline with strong theoretical and empirical grounding; or, equivalent industry research experience.

Logistics

  • Location: This role is based in San Francisco, California.

  • Compensation: Depending on background, skills and experience, the expected annual salary range for this position is $400,000 - $500,000 USD.

  • Visa sponsorship: We sponsor visas. While we can't guarantee success for every candidate or role, if you're the right fit, we're committed to working through the visa process together.

  • Benefits: Thinking Machines offers generous health, dental, and vision benefits, unlimited PTO, paid parental leave, and relocation support as needed.

How we rate this

Research, General Agents at Thinking Machines Lab rates 92 out of 100 for how much of the daily work is AI. That makes it Builds AI (AI Level 4 of 4). The level is about AI in the job, not seniority.

Classification

Builds AI. The job is building AI systems.

  1. ●●●● Builds AI80 to 100
  2. ●●●○ Works on AI60 to 79
  3. ●●○○ Uses AI40 to 59
  4. ●○○○ Little AI0 to 39

Levels come from how often the tools, models and workflows of the role are named in the posting itself. Read the description and count the AI tools, models and workflows it names.

Prepare for this job

Built from this posting and from similar jobs on Level.

Skills and AI tools this role asks for

PyTorchTensorFlowJAX

Questions you could be asked

  1. What's a project where you used PyTorch hands-on?
  2. Walk me through how you've used TensorFlow in your day-to-day work.
  3. What are the limits of JAX that you've run into, and how did you work around them?
  4. How would you decide a model or AI system is ready to ship?
  5. Tell me about a time a model underperformed in production. How did you find out, and what did you change?

Adapt your resume

  • List these exact terms on your resume: PyTorch, TensorFlow, and JAX. An applicant tracking system matches the wording, not the idea.
  • Attach one line of real, concrete experience to at least one of them. A tool named with nothing behind it rarely survives a human read.
  • Lead with what you built, trained or shipped. This role is judged on the AI system itself, not the tools around it.

Get your CV reviewed before you apply

Send your CV and the role you want next. Your review arrives by email within 3 days.

Get new research jobs (Builds AI ●●●●) by email

One email a week, each job rated for how much AI is in the work. No recruiter spam, unsubscribe in one click.

Free.

Similar roles

Research roles that build AI, at other companies.

What kind of AI work fits you?

Answer 12 practical questions in about three minutes. Get a simple profile, the work it points to, and live roles to explore next.

Find my next step

More jobs at Thinking Machines Lab

Related searches

Same AI level

Jobs by city