JAX jobs
JAX is an in-demand capability in the Gen AI and Agentic AI space. Browse 16 open roles requiring JAX on TopGenAIJobs.
Senior Quantitative Scientist – Grid Behavioral Modeling
Siemens
Bengaluru 10-12 years
Reinforcement Learning
Multi-Agent Systems
Physics-Informed Machine Learning
Game Theory
Mechanism Design
AI Solutions Developer for Cybersecurity
BNP Paribas
Chennai 4+ years
Full-time
LangChain
Gen AI
LLM
Agentic Systems
Prompt Engineering
Principal Open Source AI/ML Solutions Engineer
AMD
Bengaluru 18+ years
Full-time
C++
Python
AI/ML Frameworks
GPU kernel programming
CUDA
Deep Learning Engineer
eClerx
Mumbai / Pune / Chandigarh
Full-time
LLM
RAG
LangChain
LlamaIndex
Python
AI Engineer IV
Premera Blue Cross
United States 5+ years
Full-time
Transformers
Diffusion Models
RAG
LLM
Python
Remote ML Engineer
Turing
Pune 3+ years
Contract
Machine Learning
Python
Model training
Model evaluation
Inference pipelines
Remote ML Engineer
Turing
Delhi 3+ years
Contract
Machine Learning
Python
PyTorch
TensorFlow
JAX
Remote ML Engineer
Turing
Noida 3+ years
Contract
Machine Learning
Python
Model training
Model evaluation
Inference pipelines
AI Model Development Engineer
ETS
Not specified 3+ years
Full-time
LLM
RAG
LangChain
Gen AI
Transformer Architectures
Staff Machine Learning Engineer, AI Generation Engine
SandboxAQ
United States 8+ years
Full-time
Python
PyTorch
TensorFlow
JAX
NumPy
AI Applied Scientist
Figma
San Francisco / New York / CA / NY 6+ years
Full-time
LLM
Gen AI
Hugging Face
Reinforcement Learning
PyTorch
Machine Learning Engineer (MLE Bench)
Turing
Kenya / Egypt 3+ years
Part-time
Machine Learning
Python
PyTorch
TensorFlow
JAX
Generative AI Developer
Capgemini
Bengaluru
Full-time
LLM
Gen AI
Gemini
Python
TensorFlow
Senior Software Engineer, AI/ML
Hyderabad 5+ years
Full-time
Python
TensorFlow
JAX
Scikit-learn
Speech/audio
Senior AI Engineer
Esko
Bengaluru 6+ years
Full-time
LangChain
LlamaIndex
RAGAS
LLM-as-a-Judge
LangGraph
AI Engineer
Citi
Pune 3+ years
Full-time
LLM
Gen AI
Python
PyTorch
JAX
Get notified about new JAX jobs roles
Be the first to know when new positions matching "JAX jobs" are posted.