companyThinking Machines Lab logo

Software Engineer, Research Acceleration at thinkingmachines | San Francisco

On-site Full-time $350K/yr - $475K/yr

Clicking Apply Now takes you to AutoApply where you can tailor your resume and apply.


Unlock Your Potential

Generate Job-Optimized Resume

One Click And Our AI Optimizes Your Resume to Match The Job Description.

Is Your Resume Optimized For This Role?

Find Out If You're Highlighting The Right Skills And Fix What's Missing

Experience Level

Mid to Senior

Qualifications

Minimum Qualifications:Bachelor's degree or equivalent experience in computer science, engineering, machine learning, or a related field. Strong foundational knowledge in software engineering with a proven history of building reliable, maintainable systems. Proficiency in at least one backend programming language (Python or Rust preferred). Comfortable working across the technology stack and managing projects from inception to completion. Experience in highly collaborative settings that involve various cross-functional partners and subject matter experts. Preferred Qualifications: We encourage you to apply even if you meet some but not all of the preferred qualifications.

About the job

At Thinking Machines Lab, our mission is to empower humanity by advancing collaborative general intelligence. We aspire to create a future where everyone can access the knowledge and tools necessary to harness AI for their individual needs and aspirations.

Our team consists of scientists, engineers, and innovators who have developed some of the most renowned AI products, including ChatGPT and Character.ai, as well as open-weight models such as Mistral. We are also contributors to popular open-source initiatives like PyTorch, OpenAI Gym, Fairseq, and Segment Anything.

About the Role

We are seeking talented engineers to develop the libraries and tools that will expedite research at Thinking Machines. You will take charge of our internal infrastructure, which includes evaluation libraries, reinforcement learning training libraries, and experiment tracking platforms, all aimed at enhancing research velocity over time.

This position emphasizes collaboration; you will engage directly with researchers to pinpoint bottlenecks and challenges. Your success will be measured by the trust researchers place in your systems and their enjoyment of using them.

What You'll Do

  • Design, develop, and manage research infrastructure, including evaluation frameworks, RL training systems, experiment tracking platforms, visualization tools, and shared utilities.
  • Create high-throughput, scalable pipelines for distributed evaluation, reward modeling, and multimodal assessments.
  • Establish systems for reproducibility, traceability, and stringent quality control throughout research experiments and model training processes. Implement monitoring and observability.
  • Collaborate closely with researchers to identify obstacles and unlock new capabilities. Manage research tools like a product manager, actively seeking feedback and tracking user adoption.
  • Work alongside infrastructure, data, and product teams to ensure seamless integration of tools across the technical stack.

About Thinking Machines Lab

Thinking Machines is at the forefront of AI innovation, dedicated to creating tools and technologies that empower users. Our collaborative environment fosters creativity and progress, making us a leader in the AI field.

Similar jobs

Tailoring 0 resumes

We'll move completed jobs to Ready to Apply automatically.