481 jobs for JAX

On-device ML Performance Engineer, Graphics, Games and Machine Learning USD 166K-296K

Apple Neural Engine | C++ | CPU | Compilers | Computer Architecture

Mid-level Full Time

Seattle

2d ago

On-Device ML Infrastructure Engineer (CoreML Runtime), Graphics, Games and Machine Learning USD 170K-296K

C++ | Core ML | GPU Programming | JAX | LLVM

Mid-level Full Time

Cupertino

2d ago

Staff Tech Lead, Machine Learning Engineer, Perception USD 251K-310K

C plus plus | Computer Vision | Data pipeline | JAX | Machine Learning

Annual bonus | Company benefits program | Equity incentive plan

Senior-level Full Time

Mountain View, California, United States; San …

2d ago

AI/ML Performance Analysis and Validation Engineer USD 170K-300K

Computer Architecture | CoreML | Data center | Data center architecture | Distributed Computing

Mid-level Full Time

Cupertino

2d ago

Research Engineer, Robot Learning USD 125K-175K

Computer Vision | Data Pipelines | Debugging | Imitation Learning | JAX

Full-time role | On-site work | Work directly with founders

Mid-level Full Time

San Francisco, CA, US

2d ago

MLOps Research Engineer USD 200K-235K

AWS | Active Learning | Automated reporting | Azure | CI/CD

Mid-level Full Time

San Francisco Bay Area

2d ago

Robotics AI Engineer A SGD 162K-270K

Control Theory | Domain Adaptation | Domain Randomization | Imitation Learning | IsaacGym

Fast iteration cycle | High visibility projects | Open source contributions

Mid-level Full Time

Singapore

2d ago

AI Researcher USD 200K-300K

ASIC | C++ | Deep learning | FPGA | GPU Computing

Dental insurance | Life insurance | Medical insurance | Paid Holidays | Paid time off

Entry-level Full Time

London, United Kingdom; New York, NY, …

2d ago

ML Operations Engineer (AI/LLM) - Mercari JPY 8600K-9700K

BigQuery | Bigtable | CI/CD | Docker | Drift Detection

Flexible work hours

Senior-level Full Time

Minato City, Tokyo, Japan

3d ago

Senior Researcher - Edge AI Optimization/Hardware-Aware ML CAD 106K-164K

ARM NEON | Architecture Search | Benchmarking | C# | C++

Mid-level Full Time

Edmonton, Alberta, Canada

3d ago

Senior Robotics Engineer - R&D USD 175K-225K

Computer Vision | Diffusion Models | JAX | Language Models | Large Language Models

401k match | Dental insurance | Learning and development support | Medical insurance | Paid sick leave

Senior-level Full Time

Boston, Massachusetts

3d ago

Principal Machine Learning Engineer

Data pipeline | Deployment | Fine Tuning | GPU Computing | JAX

Senior-level Full Time

Indonesia R

3d ago

Lead Engineer, Machine Learning GBP 100K-173K

Data pipeline | Deployment | Fine Tuning | GPU | JAX

Senior-level Full Time

United Kingdom R

3d ago

Lead Engineer, Machine Learning KRW 95000K-120000K

Data pipeline | Deep learning | Fine Tuning | GPU | JAX

Senior-level Full Time

Seoul, Korea R

3d ago

Lead Engineer, Machine Learning SGD 149K-220K

Data pipeline | Deep learning | Fine Tuning | GPU Computing | JAX

Senior-level Full Time

Singapore R

3d ago

Lead Engineer, Machine Learning CNY 360K-500K

Cost Optimization | Data Pipelines | Fine Tuning | GPU Optimization | Inference Systems

Senior-level Full Time

China R

3d ago

Lead Engineer, Machine Learning

Data pipeline | Deep learning | Fine Tuning | GPU Computing | JAX

Senior-level Full Time

Indonesia R

3d ago

Staff Machine Learning Engineer

Cost Optimization | Data pipeline | Deployment | Fine Tuning | GPU Computing

Senior-level Full Time

Indonesia R

3d ago

Technical Lead, Machine Learning

Cost Optimization | Data pipeline | Fine Tuning | GPU | JAX

Senior-level Full Time

Indonesia R

3d ago

Lead Engineer, Machine Learning USD 176K-250K

Data pipeline | Fine Tuning | GPU | JAX | Latency optimization

Senior-level Full Time

United States R

3d ago

Sr. AI / ML Research Engineer-1 INR 3000K-4580K

Bayesian optimization | Computational Fluid Dynamics | Control Theory | Data Assimilation | Deep learning

Senior-level Full Time

Bangalore IBC Knowledge Park, India

4d ago

Senior Solutions Architect, Robotics Foundation Model Training USD 152K-241K

Checkpointing | Data Pipelines | Data Processing | Data loading | Data parallelism

Senior-level Full Time

US, CA, Santa Clara R

4d ago

AI Research Engineer USD 100K-150K

Accelerator hardware | Computer Vision | Data Validation | Data labeling | Deep learning

Senior-level Full Time

United States - Remote R

4d ago

Senior Robotics Engineer - Robot Perception USD 149K-204K

3D Mapping | Automated Labeling | C# | C++ | Computer Vision

Bi-weekly expense reimbursement | Travel up to 10 percent

Senior-level Full Time

USA Wilmington, MA - HQ, United …

4d ago

AIML - Foundation Model Post-Training Research Scientist CHF 164K-305K

Agentic AI | Decision Making | Deep learning | Generative Modeling | JAX

Mid-level Full Time

Zurich

4d ago

Research Scientist / Software Engineer, LLM/Agent Platform - TikTok - Content Ecology AI Innovation & Platform USD 160K-230K

Accelerator optimization | Deep learning | DeepSpeed | Distributed Systems | Evaluation Frameworks

Senior-level Full Time

San Jose, California, United States

4d ago

Member of Technical Staff - Inference USD 184K-287K

CUDA | Distributed Systems | Inference Optimization | JAX | Language Models

Paid coding challenge | Remote work options

Senior-level Full Time

San Francisco, CA; New York City, … R

5d ago

Research Scientist - Salesforce AI Research USD 117K-313K

AI Reasoning | AI Reasoning and Planning | Agentic AI | Computer Vision | Deep learning

401k | Dental insurance | Employee stock purchase program | Life and disability insurance | Medical insurance

Entry-level Full Time

California - Palo Alto, United States

5d ago

Machine Learning/AI Scientist PhD Intern, Winter 2027 USD 80K-220K

C++ | Causal Discovery | Causal Inference | Computer Vision | DDP

401k retirement plan | Health plans | Mental health support | Onsite work | Paid internship

Entry-level Full Time Internship

Los Gatos, United States

5d ago

AI Scientist USD 100K-100K

Accelerator hardware | Deep learning | Distributed Training | JAX | Language Models

Career growth | Remote work

Senior-level Full Time

United States - Remote R

5d ago

Machine Learning Research Engineer USD 100K-150K

Ablation Studies | Accelerator hardware | Data Quality | Data Validation | Data labeling

Remote work

Senior-level Full Time

United States - Remote R

5d ago

Machine Learning Engineer, Human Centered AI - Evaluations & Insights USD 170K-296K

Direct Preference Optimization | Evaluation Frameworks | Fine Tuning | Hallucination detection | Hugging Face

Mid-level Full Time

Seattle

5d ago

大模型算法工程师（数据方向） CNY 25K-37K

Ablation Testing | Apache Spark | Benchmarking | Classifier Based Filtering | Cloud Computing

Entry-level Full Time

Haidian, Beijing, China

5d ago

Senior Machine Learning Engineer USD 208K-263K

Active Learning | Birds Eye View | C++ | CUDA | Cameras

401k | Commuter benefits | Dental insurance | Disability insurance | Equity

Senior-level Full Time

San Francisco, CA

5d ago

Research Scientist, World Models, DeepMind GBP 150K-215K

CUDA | Dataset generation | Diffusion Models | GPU | Generative Models

Mid-level Full Time

London, UK

5d ago

Research Engineer, Serving Quality, DeepMind USD 174K-252K

C++ | JAX | Machine Learning | Python | Statistics

Senior-level Full Time

Mountain View, CA, USA

5d ago

AI Application Engineer TWD 2600K-3120K

C++ | Cloud platform | Computer Vision | Distributed Systems | Edge Computing

Senior-level Full Time

New Taipei, Banqiao District, New Taipei …

5d ago

Senior AI Engineer (Risk & Payments) AUD 160K-188K

AI monitoring | Agent architecture | Autogen | CrewAI | Fine Tuning

Senior-level Full Time

AU - Sydney

6d ago

Senior AI Engineer (Risk & Payments) SGD 160K-240K

AI monitoring | Autogen | CrewAI | Fine Tuning | JAX

Senior-level Full Time

SG - Singapore

6d ago

Staff/Senior Staff Infrastructure & Site Reliability Engineer – Datacentre AI Engineering - Riyadh, KSA USD 158K-237K

Ansible | Bash | CI/CD | CloudWatch | Computer Vision

Child education allowance | Employee stock purchase scheme | Housing allowance | Life and medical insurance | Live Well reimbursement

Senior-level Full Time

Riyadh, Saudi Arabia

6d ago

Senior Machine Learning Engineer GBP 65K-81K

AWS | AWS SageMaker | Airflow | Azure | Docker

Flexible working | Hybrid work arrangement | Remote work flexibility

Senior-level Full Time

Gibbs Building, United Kingdom R

6d ago

AI Research Engineer USD 80K-100K

Accelerator hardware | Agentic Systems | Data Pipelines | Deep learning | Distributed Training

Senior-level Full Time

United States - Remote R

6d ago

Machine Learning Research Intern - Summer 2027 - Sydney AUD 200K-220K

Deep learning | JAX | Machine Learning | Neural Networks | Optimization

Mentorship | Networking opportunities | Travel and accommodation

Entry-level Internship

Sydney, Australia

6d ago

ML Framework (MetalLM) Engineer USD 174K-293K

C# | C++ | CUDA | Compression | Computer Architecture

Senior-level Full Time

Cupertino

6d ago

AI/ML Senior Specialist INR 3242K-5000K

BM25 | CI/CD | Chroma | Docker | Embeddings

Hybrid work environment | On-call rotation

Senior-level Full Time

Kochi, Kerala, India

6d ago

Software Engineer, AI/Machine Learning, PhD, Early Career, 2027 Start USD 147K-210K

API Development | Artificial Intelligence | C# | C++ | Computer Vision

Mid-level Full Time

Sunnyvale, CA, USA; Atlanta, GA, USA

6d ago

Software Engineer, ML Accelerators, DeepMind USD 174K-252K

C++ | Hardware Accelerators | JAX | Kernel development | Machine Learning

Mid-level Full Time

Mountain View, CA, USA; San Francisco, …

6d ago

Campus AI/ML Researcher SGD 100K-101K

C++ | Data Mining | Deep learning | JAX | Machine Learning

International student support | Open ended research opportunities | Training provided

Entry-level Full Time

Shanghai; Singapore; Hong Kong

6d ago

AI Research Engineer INR 1800K-3000K

3D Computer Vision | Active Learning | Auto annotation | BLIP-2 | C++

Mid-level Full Time

Bengaluru, Karnataka, India

7d ago

Vice President, Artificial Intelligence & Machine Learning Engineer GBP 195K-280K

API Design | C# | C++ | Compute efficiency | Data Pipelines

Education reimbursement | Family support programs | Flexible time off | Health and wellbeing resources | Hybrid work model

Executive-level Full Time

ED3 - 20 Brandon Street, Edinburgh, … R

7d ago

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