AI Jobs
AI Jobs
Post a job
Sign up Sign in

JAX Jobs in AI

Browse 26 JAX jobs in AI for Oct 2026 with roles open at XPeng Motors, Together AI, and Discord paying $104K to $850K/year. Work as a AI Research Intern – Predictive World Model, Research Intern, Frontier Agents (Summer 2027), and Research Intern, Frontier Agents (Winter 2027).

JAX is often requested for Software Engineer jobs roles.

Receive emails of JAX Jobs in AI
Share:
AI Jobs
JAX Jobs in AI
AI Engineer jobs Ansible jobs Communications Lead jobs Customer Success Manager jobs Cypress jobs Data Scientist jobs Elasticsearch jobs Engineering Manager jobs Ethernet jobs Full Stack Engineer jobs HR jobs MongoDB jobs NCCL jobs OpenTelemetry jobs Product Manager jobs Program Manager jobs RDMA jobs React jobs Rust jobs SEO jobs Sketch jobs Tableau jobs TensorRT-LLM jobs Triton jobs TypeScript jobs
Filter by benefits
🚑 Medical Insurance 🦷 Dental Insurance 🤓 Vision Insurance 💰 401k 💰 401k Matching 💰 Equity 🏖 Unlimited Vacation 🏖 Paid Time Off 🏔 Company Retreats 🖥 Home Office Budget 📚 Learning Budget 🧘 Mental Wellness Budget ✈️ Visa Sponsorship 📦 Relocation 💊 FSA 🏥 HSA
Filter by type
Remote Full Time Internship
TA
Research Intern, Frontier Agents (Winter 2027)
Together AI / San Francisco, United States / Entry
PyTorch jobs JAX jobs Reinforcement Learning jobs
$58/hr – $70/hr 28D
TA
Research Intern, Model Shaping (Winter 2027)
Together AI / Amsterdam, Netherlands / Entry
JAX jobs Reinforcement Learning jobs Salesforce jobs
$58/hr – $70/hr 28D
TA
Research Intern, Model Shaping (Summer 2027)
Together AI / Amsterdam, Netherlands / Entry
JAX jobs Reinforcement Learning jobs Salesforce jobs
$58/hr – $70/hr 28D
TA
Research Intern, Inference (Winter 2027)
Together AI / San Francisco, United States / Entry
PyTorch jobs JAX jobs CUDA jobs
$58/hr – $70/hr 28D
TA
Research Intern, Inference (Summer 2027)
Together AI / San Francisco, United States / Entry
PyTorch jobs JAX jobs CUDA jobs
$58/hr – $70/hr 28D
XM
AI Research Intern – Predictive World Model
XPeng Motors / Santa Clara, United States / Entry
PyTorch jobs JAX jobs Reinforcement Learning jobs
29D

Frequently asked questions

What is the average salary for JAX jobs in AI?

26 active listings have salaries ranging from $104K to $850K/year.

Which companies are hiring for JAX jobs in AI?

Top companies hiring include XPeng Motors, Together AI, and Discord.

Are there remote JAX jobs in AI?

Yes, 2 of 26 are remote.