403 jobs for JAX

Partner Engineer, Generative AI USD 159K-223K

AWS | Azure | Bias Mitigation | C plus plus | Cloud Platforms

Entry-level Full Time

Menlo Park, CA

20h ago

Research Scientist/Engineer, Autonomous Security, DeepMind USD 174K-252K

Adversarial ML | Benchmarking | Data Mining | Environment Design | Function Calling

Mid-level Full Time

Mountain View, CA, USA; New York, …

20h ago

Staff Software Engineer, AI-Powered GRC Automation USD 207K-300K

Cloud Platforms | Cloud platform | Continuous controls monitoring | Controls monitoring | Data Pipelines

Senior-level Full Time

Sunnyvale, CA, USA; Kirkland, WA, USA

20h ago

Director of Data Science & AI EUR 100K-110K

AWS | Airflow | Causal Inference | Cloud Computing | Data Pipelines

Executive-level Full Time

Barcelona, Spain

23h ago

ML Engineer, Surrogate Modeling (Vehicle Engineering) USD 125K-175K

Active Learning | Adaptive Sampling | CFD | Continuous integration | Data Pipelines

401k retirement plan | Employee stock purchase plan | Life insurance | Long-term disability insurance | Long-term incentives

Entry-level Full Time

Hawthorne, CA

1d ago

Lead AI Research Scientist USD 357K-357K

Agent evaluation | CUDA | Deterministic Planning | Human Feedback | JAX

Senior-level Full Time

San Francisco, California

1d ago

Mid-Level Machine Learning Engineer USD 110K-200K

ASIC | C++ | CoreML | Edge AI | Embedded Systems

Mentorship

Mid-level Full Time

Fremont , CA

1d ago

Senior Machine Learning Engineer USD 200K-280K

AI Chip SDK | AI chip | ARM Ethos | C++ | Edge AI

Senior-level Full Time

Fremont , CA

1d ago

AI Software Engineer (Vehicle Engineering) USD 125K-175K

3D Reconstruction | Agent systems | Agentic AI | Anomaly Detection | CI/CD

401k retirement plan | Dental insurance | Employee stock purchase plan | Life insurance | Life insurance coverage

Senior-level Full Time

Hawthorne, CA

1d ago

Senior Quantum AI Research Scientist, Applied Research USD 192K-304K

Adapters | CUDA | Calibration | Deep learning | Distributed Training

Senior-level Full Time

US, WA, Redmond, United States

1d ago

Senior DGX Cloud AI Infrastructure Software Engineer CNY 160K-240K

AI Inference | AI Training | APIs | C# | C++

Senior-level Full Time

China, Shanghai

1d ago

AI Research Engineer (Applied AI) USD 100K-150K

Accelerator hardware | Agentic Systems | Computer Vision | Data Quality | Data labeling

Remote work

Mid-level Full Time

United States - Remote R

1d ago

Applied Research Scientist, Multi-Modal Perception (PhD New Grad) USD 175K-215K

C++ | Computer Vision | JAX | Machine Learning | PyTorch

Entry-level Full Time

Mountain View, CA USA; San Francisco, …

1d ago

Machine Learning Engineer (Training Optimization) CNY 144K-240K

CUDA | DeepSpeed | Diffusion Models | Distributed Training | FSDP

Entry-level Full Time

Beijing, Beijing, China

1d ago

AI Research Engineer USD 234K-385K

Cryptography | Distributed Systems | Federated Learning | Go | JAX

Mid-level Full Time

Anywhere R

2d ago

Machine Learning Engineer: Perception and Planning USD 193K-270K

Automated testing | Behavior Prediction | C++ | Code review | Computer Vision

Senior-level Full Time

Oakland, CA

2d ago

Staff Applied Scientist GBP 225K-325K

Black | Cloud Platforms | Complex Planning | Deep learning | Explainable AI

Senior-level Full Time

London

2d ago

Senior Applied Scientist GBP 105K-135K

Black | Cloud Computing | Complex Planning | Deep learning | Explainable AI

Flexible working options | Hybrid work scheme | Share options

Senior-level Full Time

London

2d ago

Machine Learning Engineer USD 150K-225K

A/B | A/B Testing | Anomaly Detection | B testing | CI/CD

Mid-level Full Time

Kathmandu

2d ago

AI Engineer - Mission Innovation Lab USD 108K-198K

Adapters | Adversarial Testing | Airflow | C++ | CI/CD

Active security clearance required | Background Investigation required | Onsite work

Senior-level Full Time

Pittsburgh, United States

2d ago

AI Research Engineer (Applied AI) USD 100K-150K

Computer Vision | Deep learning | Distributed Training | JAX | Language Models

Mid-level Full Time

United States - Remote R

2d ago

Sr GenAI Infra Specialist SA, AWS WWSO Startup USD 153K-228K

AWS | Amazon EC2 | Amazon EKS | Amazon S3 | Cache optimization

Inclusive team culture | Mentorship and career growth | Work-life balance

Senior-level Full Time

New York, New York, USA

2d ago

Student Assistants in Intelligent Connected Robotics EUR 30K-34K

3D Printing | Actuator integration | Autonomy | C# | C++

Flexible working hours | Onsite prototyping experience | Part-time option

Entry-level Full Time

Berlin, DE, 10587

2d ago

Data Scientist - Graduate USD 84K-147K

Classification | Confidence Intervals | Convolutional Neural Networks | Data Analysis | Deep learning

401k employer match | Disability insurance | Education assistance | Employee stock purchase program | Gym membership discounts

Entry-level Full Time

Boston, United States

2d ago

Inference Intern - Summer 2027 USD 64K-142K

Accelerator architectures | C++ | Compilers | Debugging | Distributed Systems

Daily Dinner | Daily lunch | Direct mentorship | Housing support | Paid internship

Entry-level Internship

San Jose

2d ago

Inference Intern - Spring 2027 USD 60K-142K

C++ | Compilers | Consensus Protocols | Consistency models | Distributed Systems

Daily Dinner | Daily lunch | Direct mentorship | Housing support | Paid internship

Entry-level Internship

San Jose

2d ago

AI Research Intern

Computer Vision | Debugging | Deep learning | GPU Training | Imitation Learning

Entry-level Internship

Zürich

2d ago

Director of Data Science & AI EUR 100K-116K

AWS | Airflow | Causal Inference | Deep learning | GCP

Executive-level Full Time

Barcelona, Spain

2d ago

Research Scientist - Machine Learning (Contractor) GBP 59K-62K

AI Agents | Bayesian optimization | JAX | Language Models | Large Language Models

Mid-level Contract

London, United Kingdom

2d ago

Machine Learning Specialist (Research and Engineering) PHP 130K-175K

API Development | AWS | Azure | CI/CD | Data Augmentation

Mid-level Full Time

Taguig, Metro Manila, Philippines

2d ago

Machine Learning Engineer (Agentic AI)

Agent Orchestration | Agent systems | Autogen | Automated Evaluation | Benchmarking

Senior-level Full Time

Seoul HQ

3d ago

Generative AI Developer INR 1400K-1700K

Cloud Functions | Cloud Storage | Cloud platform | Compute Engine | Gemini

Digital learning platform access | Elder care | Flexible work | Health checks | Insurance with top-ups

Mid-level Full Time

Bangalore, IN

3d ago

AI/ML Engineer USD 110K

AWS | Agentic AI | Apache Spark | CI/CD | Cloud Run

401k | Employee assistance program | Family leave | Life insurance | Medical insurance

Mid-level Full Time

NY-New York;1301 Avenue of the Americas …

3d ago

AI/ML Engineer USD 110K

AWS | Agentic AI | Apache Spark | CI/CD | Cloud Run

Mid-level Full Time

NY-New York;1301 Avenue of the Americas …

3d ago

IN_Senior Manager_AI Engineer_Digital Engineering Transformation_Advisory_Bangalore INR 2000K-4512K

Amazon Bedrock | Anthropic API | Azure Machine Learning | Azure OpenAI | Chroma

Senior-level Full Time

Bengaluru Millenia, India

3d ago

AI Software Engineer, Kernel Libraries - New College Grad 2026 USD 124K-241K

Apache TVM | C++ | CUDA | CuTile | FlashAttention

Employee benefits | Equity

Senior-level Full Time

US, CA, Santa Clara, United States

3d ago

大模型训练与推理Infra工程师-MiMo CNY 500K-500K

C plus plus | CUDA | CUDNN | DeepSpeed | Docker

Mid-level Full Time

北京 R

3d ago

AI engineer USD 152K-315K

AWS | Azure | Cloud Computing | Deep learning | GCP

401k | Company offsites | Dental insurance | Fertility support | Flexible spending account

Mid-level Full Time

San Francisco, CA

3d ago

AI engineer (UK) GBP 80K-90K

AWS | Bias detection | Cloud platform | Deep learning | Explainability

Company off-sites | Dental insurance | Fertility and family planning support | Learning and development stipend | Medical insurance

Mid-level Full Time

London, UK

3d ago

AI Research Scientist - Machine Learning for Behavioral AI USD 174K-250K

3D conformers | ASE | Agent systems | Alignment | Autoregressive models

Mid-level Full Time

Pasadena, CA

3d ago

Research Engineer, Responsibility Engineering, DeepMind GBP 250K-340K

Automated testing | Evaluation Frameworks | Inference pipeline | JAX | Language Processing

Senior-level Full Time

London, UK

3d ago

Director of Data Science & AI EUR 100K-110K

AWS | Apache Airflow | Causal Inference | Deep learning | Event Stream

Executive-level Full Time Part Time

Barcelona, Spain

4d ago

AI Software Lead – PyTorch & CUDA Runtime (Next-Gen Accelerator) INR 2475K-3380K

C# | C++ | CUDA | Compiler runtime interaction | Compiler/runtime

Senior-level Full Time

Bengaluru, KA, India

4d ago

AI SW Stack Deployment Architect INR 2500K-4500K

API Design | Cloud Computing | Distributed Systems | Edge Computing | Inference Server

Senior-level Full Time

Bengaluru, KA, India

4d ago

AI Research Engineer (Applied AI) USD 150K-222K

Accelerator hardware | Agentic Systems | Data Quality | Data quality monitoring | Deep learning

Career growth | Remote work

Mid-level Full Time

United States - Remote R

4d ago

Senior Machine Learning Engineer USD 151K-265K

API Design | AWS | Azure | Cloud Platforms | Context engineering

Senior-level Full Time

San Jose, United States R

4d ago

Applied AI Engineer USD 99K-225K

AWS | AgentOps | Azure | ChromaDB | Continued Pretraining

401k retirement plan | Bike storage | Commuter benefits | Dependent care FSA | Desk setup stipend

Mid-level Full Time

Washington DC

4d ago

Senior Simulation Engineer - Machine Learning USD 140K-230K

Ablation Studies | C++ | CI/CD | Continuous Deployment | Continuous integration

Senior-level Full Time

Palo Alto, CA

4d ago

Senior Machine Learning Scientist USD 151K-235K

Contrastive Learning | Deep learning | Diffusion Models | Distributed machine learning | Graph Neural Networks

Dental insurance | Health insurance | Lunch budget | Office setup budget | Paid time off

Senior-level Full Time

South San Francisco

4d ago

ML Scientist USD 200K-250K

Agentic AI | Context modeling | Deep learning | Distributed Training | Information Extraction

In person work in Boston | Stock options

Mid-level Full Time

Boston or NYC

4d ago

Find jobs in AI/ML, Data Science and Big Data