Filter By


Save Search

MORE FILTERS

LESS FILTERS

  • Job Types

396 Jobs in JAX framework

Research, Post-Training

Thinking Machines Lab

San Francisco, CA, USA

0-2 year

___
Full Time

25 Nov 2025

Summary

About the Role

The role of post-training researchers sits at the core of our roadmap. This is the critical bridge between raw model intelligence and a system that is actually useful, safe,...

Key Skills

Deep Learning,JAX framework,Large Language Models - LLMs,Python Programming,PyTorch,TensorFlow

___
Full Time

25 Nov 2025

Summary

About the Role

Thinking Machines builds multimodal-first. For us, there is no separate multimodal work. It’s at the core of everything we do, from the scientific goals we’re setting to...

Key Skills

Effective communication skills,JAX framework,Machine learning techniques,PyTorch,Real time scoring models,TensorFlow

Principal Software Engineer, ML Frameworks

Tenstorrent Inc.

Austin, TX, USA; Santa Clara, CA, USA

6-8 year

___
Full Time

7 Nov 2025

Summary

During the interview process, candidates will be assessed for the appropriate level, and offers will align with that level, which may differ from the one in this posting.

Who You...

Key Skills

C++,Design,JAX framework,Machine learning techniques,PyTorch,TensorFlow

___
Full Time

7 Nov 2025

Summary

During the interview process, candidates will be assessed for the appropriate level, and offers will align with that level, which may differ from the one in this posting.

Who You...

Key Skills

C++,Design,Infrastructure as code,JAX framework,Machine learning techniques,PyTorch,TensorFlow

___
Full Time

14 Oct 2025

Summary

We welcome candidates at various experience levels. During the interview process, candidates will be assessed for the appropriate level, and offers will align with that level, which may differ...

Key Skills

Aartificial intelligence,C++,JAX framework,Machine learning techniques,Python Programming,PyTorch,TensorFlow

___
Full Time

14 Oct 2025

Summary

We welcome candidates at various experience levels for this role. During the interview process, candidates will be assessed for the appropriate level, and offers will align with that level, which...

Key Skills

Aartificial intelligence,C++,Infrastructure as code,JAX framework,Machine learning techniques,Optimization,Python Programming,PyTorch,TensorFlow

___
Full Time

14 Oct 2025

Summary

We welcome candidates at various experience levels for this role. During the interview process, candidates will be assessed for the appropriate level, and offers will align with that level, which...

Key Skills

Aartificial intelligence,C++,JAX framework,Machine learning techniques,Python Programming,PyTorch,TensorFlow

___
Full Time

14 Oct 2025

Summary

Who You Are

  • A passionate software engineer eager to work on compiler technologies and the challenges of AI hardware, whether from compilers, systems, or broader software...

Key Skills

Aartificial intelligence,C++,JAX framework,Machine learning techniques,Python Programming,PyTorch,TensorFlow

___
Full Time

14 Oct 2025

Summary

Who You Are

  • A passionate software engineer eager to work on compiler technologies and the challenges of AI hardware, whether from compilers, systems, or broader software...

Key Skills

Aartificial intelligence,C++,JAX framework,Machine learning techniques,Python Programming,PyTorch,TensorFlow

Sr. Engineer, Software - AI Compiler

Tenstorrent Inc.

Austin, TX, USA; Santa Clara, CA, USA

6-8 year

___
Full Time

14 Oct 2025

Summary

Who You Are

  • A passionate software engineer eager to work on compiler technologies and the challenges of AI hardware, whether from compilers, systems, or broader software...

Key Skills

Aartificial intelligence,Analytical and Problem solving,C++,JAX framework,Machine learning techniques,Python Programming,PyTorch,TensorFlow