Sign Up Sign In

Post a Job

Post a job

Sign up Sign in

Full Time JAX AI Jobs in San Jose, California, United States

Browse 2 AI jobs in San Jose for Nov 2026 with roles open at Samsung Semiconductor, Inc. and SambaNova paying $131K to $253K/year. Work as a Staff Engineer, ML Profiling Tools and AI Performance Engineer - New Grad.

JAX is often requested for Software Engineer jobs roles.

Receive emails of Full Time JAX AI Jobs in San Jose, California, United States

Ansible Bash CI/CD CUDA Compiler Engineer Datadog Ethernet JAX Kubernetes MongoDB NCCL Platform Engineer Product Manager Prototyping Python RDMA Ray React Reinforcement Learning Rust Tableau Technical Documentation TensorRT-LLM Triton TypeScript vLLM

Cities

San Francisco 5 Bristol 4 New York 3 San Jose Cambridge 2 London 2 Washington 1

Countries

United States 12 California 9 United Kingdom 6 New York 3 India 1

Filter by type

Full Time

S

AI Performance Engineer - New Grad

SambaNova / Austin, TX ; San Jose, CA / Entry

PyTorch TensorFlow JAX

$131K – $197K 11D

SSI

Staff Engineer, ML Profiling Tools

Samsung Semiconductor, Inc. / San Jose, United States / Staff

TensorFlow JAX Excel

$163K – $253K 29D

Frequently asked questions

What is the average salary for Full Time JAX jobs?

2 active listings have salaries ranging from $131K to $253K/year.

Which companies are hiring for Full Time JAX jobs?

Top companies hiring include Samsung Semiconductor, Inc. and SambaNova.