415 jobs for JAX

Career Starter Data Science CHF 120K-140K

AI Engineering | CI/CD | Cloud Data | Cloud Data Warehouse | Data Analysis

Senior-level Full Time

Bern 65, CH

12h ago

AI Governance | AWQ | AWS CDK | AWS SageMaker | Agent systems

Equipment and office stipend | Flexible PTO | Learning and development stipend | Medical insurance | Paid exams and certifications

Mid-level Full Time

MEXICO R

1d ago

AI/ML Engineering Manager USD 140K-185K

AWS | AWS CDK | AWS CloudFormation | AWS Glue | Airflow

100 percent remote work | Equipment and office stipend | Flexible PTO | Generous holidays | Individual professional development plan

Mid-level Full Time

ARGENTINA R

1d ago

AI/ML Engineering Manager

AWS | AWS Glue | Airflow | Amazon Bedrock | Amazon Kinesis

100% remote work | Annual learning stipend | Equipment stipend | Flexible time off | Office stipend

Mid-level Full Time

BRAZIL R

1d ago

AI/ML Engineering Manager USD 140K-215K

AWS | AWS CDK | AWS CloudFormation | AWS Glue | Agent systems

401k plan | Company laptop | Dental insurance | Equipment and office stipend | Flexible spending account

Mid-level Full Time

USA R

1d ago

Machine Learning Lead (LLM) USD 165K-210K

Agentic Workflows | Data Analysis | Deep learning | Docker | Embeddings

Dental coverage | Health coverage | Medical coverage | Meetups in NYC and DC | Remote-first

Senior-level Full Time

Remote R

1d ago

Retrosynthesis Researcher, Machine Learning USD 120K-145K

ASKCOS | AiZynthFinder | Attention Mechanism | Cheminformatics | Cloud Computing

401k | Dental insurance | Flexible work schedule | Health care | Paid vacation

Senior-level Full Time

New York

1d ago

AI System Research and Development Engineer - Optimization USD 200K-287K

Benchmarking | CUDA | CUDNN | Cutlass | Deep learning

Mid-level Full Time

US-WA-Bellevue

1d ago

VP of Research, Machine Learning GBP 195K-280K

Agent systems | Alignment | Deep learning | GPU Computing | Guardrails

Executive-level Full Time

United Kingdom

1d ago

Staff Machine Learning Engineer GBP 90K-120K

Bias Evaluation | Data Pipelines | Direct Preference Optimization | Fine Tuning | GPU Optimization

Senior-level Full Time

United Kingdom

1d ago

Principal Machine Learning Engineer GBP 90K-130K

Bias Evaluation | DPO | Deep learning | DeepSpeed | Distillation

Senior-level Full Time

United Kingdom

1d ago

Technical Lead, Machine Learning GBP 100K-135K

Bias Evaluation | DPO | Data Pipelines | Evaluation | GPU Optimization

Senior-level Full Time

United Kingdom

1d ago

Senior Machine Learning Engineer GBP 90K-120K

Data Pipelines | Debugging | Deep learning | GPU Computing | Inference

Senior-level Full Time

United Kingdom

1d ago

Member of Technical Staff, Machine Learning GBP 90K-115K

Data Pipelines | Deep learning | Fine Tuning | GPU Computing | Inference

Senior-level Full Time

United Kingdom

1d ago

Applied AI Engineer GBP 90K-130K

Data Pipelines | Fine Tuning | Inference | JAX | LLM

Senior-level Full Time

United Kingdom

1d ago

2026 -Senior AI/ML Engineer – Multimodal Content Intelligence - Permanent EUR 78K-108K

Computer Vision | Deep learning | Fine Tuning | JAX | Language Models

Senior-level Full Time

Dublin, Ireland

1d ago

Senior Applied ML Engineer, Graph Neural Network, ML Frontiers CHF 198K-236K

Artificial Intelligence | C++ | Colab | Data Structures | Data structures algorithms

Senior-level Full Time

Zürich, Switzerland

1d ago

Research Engineer, Frontier Safety Mitigations, DeepMind GBP 225K-300K

AI Coding Agents | AI coding | Adversarial Machine Learning | Anomaly Detection | Coding Agents

Mid-level Full Time

London, UK

1d ago

AI Solution Architect | Gcore | Remote A PLN 237K-400K

Ansible | CI/CD | Distributed Training | Distributed inference | Docker

Extra paid vacation and sick leave | Flexible working hours | Language courses | Private medical insurance | Remote work

Senior-level Full Time

Remote but only within Poland R

1d ago

Large Model Algorithm Researcher SGD 100K-170K

AI Agents | Artificial Intelligence | JAX | PyTorch | Reinforcement Learning

Entry-level Full Time

Singapore-CapitaSky

1d ago

Senior Machine Learning Engineer, End‑to‑End Autonomous Driving USD 184K-356K

Active Learning | CI/CD | Curriculum learning | Data Pipelines | Data Preprocessing

Equity | Health insurance

Senior-level Full Time

US, CA, Santa Clara, United States

1d ago

Senior AI Compiler Engineer, Algorithms and Code-Generation USD 152K-241K

C# | C++ | CPU architecture | CUDA | Deep learning

Senior-level Full Time

US, CA, Santa Clara, United States

1d ago

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

C++ | CUDA | CUDNN | Context caching | DeepSpeed

Entry-level Full Time

北京 R

1d ago

ML Engineer -Signal Processing / ASR INR 2500K-4500K

Audio classification | Automatic Speech Recognition | CPU Utilization | Cost Optimization | Data Analysis

Mid-level Full Time

India (Remote) R

2d ago

Senior Staff Machine Learning Engineer, Content Platform GBP 72K-85K

Audio Processing | Classification | Computer Vision | Data Quality | JAX

Work from home flexibility

Senior-level Full Time

London

2d ago

Senior Software Engineer, Performance Optimizations for LLM USD 174K-252K

C++ | Compute Optimization | Deep learning | GPU | JAX

Senior-level Full Time

Sunnyvale, CA, USA; Kirkland, WA, USA

2d ago

Senior Analyst INR 2500K-4800K

AWS | Adapter-Tuning | Azure | C++ | CI/CD

Work-from-home hybrid

Senior-level Full Time

Pune, Maharashtra, India

2d ago

Applied Scientist - Multimodal Foundation Models & Robotics

C++ | Deep learning | Diffusion Models | Imitation Learning | JAX

Mid-level Full Time

Zürich, ZH, CH

2d ago

PhD Student - Privacy Enhancing Technologies for LLMs and AI Agents EUR 33K-40K

Agent systems | Deep learning | JAX | Language Models | Language Processing

Conference publishing support | Healthy Meals | Language courses | Mentorship | Mentorship and collaboration resources

Entry-level Full Time

München, Germany

2d ago

Staff Software Engineer, Gen AI USD 195K-245K

CNN | Computer Vision | Diffusion Models | Docker | Fine Tuning

401k match | Adoption leave | Cell phone reimbursement | Discounted pet insurance | Electronics reimbursement

Senior-level Full Time

Mountain View, California, United States

2d ago

Principal Applied Scientist- AI USD 200K-290K

Cloud Computing | Computer Vision | Data benchmarking | Deep learning | JAX

Hybrid work flexibility | Reasonable accommodations | Remote work options

Senior-level Full Time

Bellevue

2d ago

AI Research Engineer USD 250K-386K

Applied cryptography | Cryptography | Deep learning | Distributed Systems | Federated Learning

Entry-level Full Time

Anywhere R

2d ago

Machine Learning Engineer: Perception and Planning USD 184K-264K

Automated testing | Behavior Prediction | C plus plus | Classification | Code review

Senior-level Full Time

Oakland, CA

2d ago

Senior Machine Learning Engineer, Runtime and Serving USD 213K-263K

Benchmarking | Buffer management | C++ | CUDA | Concurrent Systems

Senior-level Full Time

Mountain View, CA, USA

2d ago

Senior AI Software Architect - Runtime USD 195K-361K

Benchmarking | C# | C++ | CI/CD | CUDA

Hybrid work model

Senior-level Full Time

USA - OR - Hillsboro, United …

2d ago

Senior Deep Learning Compiler Verification Engineer USD 140K-224K

C++ | Formal verification | Graph optimization | IR lowering | JAX

Comprehensive benefits package | Equity

Senior-level Full Time

US, CA, Santa Clara, United States

2d ago

Applied Scientist USD 80K-218K

AWS | Agentic Systems | Artificial Intelligence | Azure | Cloud Computing

Employee assistance program | Employee stock purchase plan | Fitness reimbursement | Flexible work arrangements | Hybrid work model

Mid-level Full Time

United States of America, Eagan, Minnesota R

2d ago

Vision Foundation Model Research Intern USD 114K-114K

Computer Vision | Debugging | Deep learning | Distributed Training | JAX

Bonus | Equity | Health benefits

Entry-level Internship

Mountain View, California

3d ago

Senior Machine Learning Research Scientist USD 200K-220K

AWS SageMaker | Apache Airflow | CUDA | Data parallelism | Distributed Training

401k match | Medical/Dental/Vision insurance | Paid Holidays | Paid parental leave | Remote-first team

Senior-level Full Time

Remote (United States) R

3d ago

AI/ML Developer INR 1000K-1890K

AWS SageMaker | Accelerate | Adapter Layers | Apache Spark | Azure Machine Learning

Mid-level Full Time

Gurugram, India

3d ago

Research Engineer, Pretraining, DeepMind USD 174K-252K

Fine Tuning | JAX | Language Models | Large Language Models | Machine Learning

Mid-level Full Time

New York, NY, USA

3d ago

Senior Customer Engineer, AI Infrastructure, Google Cloud SGD 160K-200K

AI Accelerator Hardware | AI accelerator | AI compute | AI compute clusters | AI infrastructure

Senior-level Full Time

Singapore

3d ago

SOC Architect, XProf USD 147K-211K

C# | C++ | Compiler profiling | Data Analysis | Data Visualization

Senior-level Full Time

Sunnyvale, CA, USA

3d ago

Research Engineer, Deployment Performance, Robotics, DeepMind GBP 100K-180K

C++ | JAX | Machine Learning | Performance Tuning | Profiling

Mid-level Full Time

London, UK

3d ago

Principal Software Engineer, AI Agents & Cloud Integration INR 2500K-4500K

CI/CD | Docker | Git | Graph Database | JAX

Hybrid work model

Senior-level Full Time

Bangalore, India

3d ago

Staff Scientist – AI for Science USD 94K-147K

C# | C++ | Data Mining | Data Science | Distributed Training

Collaborative research environment | Comprehensive benefits | Conference presentations | Hybrid work up to 40 percent remote | Inclusive professional growth

Senior-level Full Time

Lemont, IL USA, United States

3d ago

Applied Reinforcement Learning Engineer 2 USD 150K-300K

ActorCritic | BCQ | BehavioralCloning | CQL | DQN

Mid-level Full Time

Redmond, Washington, United States

3d ago

Principal High-Performance LLM Training Engineer USD 272K-431K

Activation checkpointing | Benchmarking | CUDA | Communication and Computation Overlap | Compilers

Benefits | Equity

Senior-level Full Time

US, CA, Santa Clara, United States

3d ago

Sr Gen AI Engineer USD 87K-140K

ARM Templates | Ansible | Apache Spark | Azure | Azure AD

401k matching | Corporate fitness program | Flexible time off | Inclusion and belonging | Medical, dental, and vision insurance

Senior-level Full Time

Dallas, United States

3d ago

Machine Learning Research Engineer - Mellon College of Science - Pittsburgh Supercomputing Center USD 119K-206K

Benchmarking | Containerization | Continuous Testing | Data analytics | Django

Dental insurance | Fitness center access | Life and accidental death and disability insurance | Medical insurance | Observed holidays

Mid-level Full Time

Pittsburgh, United States

3d ago

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