310 jobs for JAX

Machine Learning Engineer, Foundation Model USD 129K-247K

Auto-regressive models | C plus plus | Deep learning | Diffusion Models | Distributed Training

Senior-level Full Time

San Jose

13h ago

Senior AI Engineer PHP 100K-160K

AI Pipelines | Agent Builder | Artifact Registry | Automated retraining | BigQuery

Senior-level Full Time

México (Remote) R

17h ago

Senior Software Engineer, ML Performance Infrastructure USD 174K-252K

C# | C++ | Compute technology | Data Storage | Data Structures

Senior-level Full Time

Sunnyvale, CA, USA

23h ago

Senior Software Engineer, AI Training & Infrastructure USD 200K-300K

AWS | Alerting | Algorithms | Automated testing | Azure

Senior-level Full Time

San Mateo

1d ago

Staff Software Engineer, Machine Learning Compilers, Edge TPU USD 207K-300K

Accelerator development | Co-design | Compilation Optimization | Compiler development | Compute architecture

Senior-level Full Time

Mountain View, CA, USA; Kirkland, WA, …

1d ago

Staff Software Engineer, Generative AI, Core ML USD 207K-300K

AI Feedback | Computer Vision | Data Processing | Deep learning | Digital Twin

Senior-level Full Time

Mountain View, CA, USA

1d ago

Senior Staff Software Engineer, Machine Learning, ML Training USD 262K-365K

Cloud Computing | Cloud TPU | Cloud platform | Data Processing | Debugging

Senior-level Full Time

Kirkland, WA, USA; Seattle, WA, USA

1d ago

PhD scholarship in large-scale Bayesian deep learning - DTU Compute DKK 428K-428K

Approximate Inference | Bayesian Inference | Bayesian deep learning | Deep learning | Differential geometry

Academic freedom | Collegial respect | Flexible work-life balance | Free seminars | International environment

Entry-level Full Time

Kgs. Lyngby, Denmark

2d ago

AI Research Engineer USD 295K-425K

Cryptography | Deep learning | Distributed Fine Tuning | Distributed Systems | Federated Learning

Mid-level Full Time

Anywhere R

2d ago

AI Scientist - Machine Learning (US/KR) USD 136K-185K

Ablation Studies | Benchmarking | Data Analysis | Deep learning | Exploratory Data Analysis

Senior-level Full Time

Palo Alto, CA / Yeoksam, Seoul R

2d ago

Staff Applied Scientist GBP 225K-325K

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

Hybrid work | Remote work flexibility | Stock options

Senior-level Full Time

London

2d ago

Senior Applied Scientist GBP 225K-325K

Black | Cloud Platforms | Complex Planning | Data Analysis | Deep learning

Hybrid work

Senior-level Full Time

London

2d ago

Machine Learning Engineer II CAD 154K-199K

C# | C++ | Data Engineering | Data platform | Deep learning

Entry-level Full Time

661 University Avenue, Toronto, Ontario, Canada

2d ago

Research Machine Learning Scientist II CAD 150K-190K

AWS | Azure | Cloud Computing | Data Processing | Deep Learning Training

Senior-level Full Time

661 University Avenue, Toronto, Ontario, Canada

2d ago

Senior High-Performance LLM Training Engineer USD 184K-356K

C++ | CUDA | Computer Architecture | Deep learning | GPU Architecture

Senior-level Full Time

US, CA, Santa Clara, United States

2d ago

Senior Robotics Research Engineer, Robotics and AI for Drug Discovery USD 184K-356K

Agentic AI | Artificial Intelligence | Bimanual manipulation | C++ | CI/CD

Senior-level Full Time

US, CA, Santa Clara, United States

2d ago

Postdoc Research Associate in Machine Learning USD 61K-67K

JAX | Machine Learning | Model Interpretability | Model Reliability | Model calibration

Defined benefit plan | Defined contribution plan | Dental insurance | Employee assistance program | Flexible spending accounts

Mid-level Full Time

HOWE - Howe Hall, United States

2d ago

Machine Learning Scientist 4 - Pricing Science USD 300K-537K

Causal Inference | Deployment | Difference-in-differences | Elasticity | Experimental Design

401k Retirement Plan Employer Match | Disability programs | Family-forming benefits | Flexible spending accounts | Flexible time off

Mid-level Full Time

USA - Remote, United States R

2d ago

Research Engineer, Information Quality USD 174K-252K

Artificial Intelligence | Artificial Intelligence Agents | Data Analysis | Data Visualization | Deep learning

Senior-level Full Time

Mountain View, California, US

2d ago

Junior AI Engineer USD 100K-120K

Agentic Workflows | CI/CD | Chunking | Context windows | Embedding Models

Entry-level Full Time

New York, New York, United States

2d ago

Senior Software Engineer, LLM Performance USD 180K-339K

C++ | CUDA | Cutlass | FlashAttention | FlashInfer

Senior-level Full Time

SF Bay Area (Hybrid) R

2d ago

Principal AI/ML Architect USD 165K-185K

AWS Lambda | Airflow | Amazon Bedrock | Amazon Kinesis | Amazon OpenSearch

401k with company match | Company issued laptop | Dental insurance | Equipment and office stipend | Flexible spending account

Senior-level Full Time

USA R

2d ago

Principal AI/ML Architect MXN 1040K-1462K

AWS CDK | AWS CloudFormation | Agentic Systems | Airflow | Amazon Bedrock

Competitive equity | Equipment stipend | Flexible time off | Learning and development stipend | Paid exams and certifications

Senior-level Full Time

MEXICO R

2d ago

Machine Learning Engineer: Perception and Planning USD 170K-230K

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

Senior-level Full Time

Oakland, CA

3d ago

Machine Learning Engineer, End-to-end Autonomy USD 112K-184K

AWS | Ablation Studies | Azure | C++ | Data Curation

Comprehensive benefits package | Flexible work environment | Relocation support

Senior-level Full Time

Palo Alto, CA

3d ago

Senior Machine Learning Engineer, End-to-end Autonomy USD 140K-230K

3D Perception | C++ | CUDA | Computer Vision | Data Preprocessing

Senior-level Full Time

Palo Alto, CA

3d ago

Senior Machine Learning Engineer, Perception USD 170K-220K

3D Geometry | 3D Perception | BEV Representation | Calibration | Camera Radar

Senior-level Full Time

Santa Clara, CA

3d ago

Senior Machine Learning Engineer USD 151K-265K

AWS | Alignment | Azure | Deep learning | Docker

Senior-level Full Time

San Jose, United States R

3d ago

Senior AI Engineer USD 150K-160K

AWS Bedrock | Agent Orchestration | Distributed Systems | Docker | Drift Detection

401k match | Health insurance | Paid time off | Remote work options

Senior-level Full Time

Redmond, Washington, United States

3d ago

Group 2-24 | Co-Op | Biomedical & Physiological Signal Processing & Machine Learning | Jun-Dec 2026 USD 50K-62K

Deep learning | High Performance | High-Performance Computing | JAX | Keras

Background Investigation required | Secret clearance eligible

Entry-level Full Time

Lexington, MA, US

3d ago

Robotics Researcher - Physical AI CAD 121K-182K

6D Pose Estimation | CUDA | Computer Vision | Dexterous Manipulation | Distributed Training

Dental insurance | Employee assistance program | Employer Paid DC Pension | Employer Savings Plan Matching | Fitness reimbursement

Senior-level Full Time

Canada-Ontario-Waterloo

3d ago

Sr Gen AI Engineer USD 87K-140K

AKS | Active Directory | Apache Spark | Automated testing | Azure

401k matching | Corporate fitness program | Flexible time off | Medical/Dental/Vision | Voluntary benefits

Senior-level Full Time

Dallas, United States

3d ago

Solutions Architect, LLM Model Builder USD 152K-241K

Benchmarking | CUDA | Compression | Distillation | Evaluation

Senior-level Full Time

US, CA, Santa Clara, United States

3d ago

Solutions Architect, LLM Model Builder USD 152K-241K

Benchmarking | CUDA | Compression | Distillation | Evaluation

Senior-level Full Time

US, CA, Santa Clara, United States

3d ago

Lead AI Engineer CAD 170K-200K

A/B | A/B Testing | APIs | AWS | Agentic Architectures

Access to training academy | Flexible work model | Mental physical financial wellbeing support | Mentorship | Professional learning program

Senior-level Full Time

300 - Toronto, Canada

3d ago

AI Engineer I USD 50K-70K

AWS Bedrock | AWS SageMaker | Axolotl | BitsAndBytes | DPO

401k match | Commuter benefits | Employee stock option program | Global recharge days | Learning and development opportunities

Entry-level Full Time

Chicago, IL

3d ago

Partner Engineer, Generative AI USD 173K-247K

AWS | Agent Orchestration | Azure | Bias Mitigation | C plus plus

Senior-level Full Time

Menlo Park, CA

3d ago

AI Scientist - Warsaw PLN 200K-247K

Deployment | Fine Tuning | Go | JAX | Java

Gympass contribution | Meal allowance | Private pension plan

Mid-level Full Time

Warsaw

4d ago

AI Scientist - Zurich CHF 98K-148K

Distributed Training | Go | JAX | Java | Kubernetes

Gympass subscription | Meal allowance | Private pension plan

Mid-level Full Time

Zurich

4d ago

Senior Staff Machine Learning Engineer USD 264K-378K

Agile | Deep learning | Experimentation | Generative AI | JAX

401k retirement plan | Flexible holidays | Health insurance | Paid parental leave | Paid sick leave

Senior-level Full Time

New York, NY

4d ago

Systems Software Engineer, AI Infrastructure INR 2000K-4500K

AWS | Automation | Azure | C++ | CI/CD

Senior-level Full Time

India, Bengaluru

4d ago

Data Scientist - Agentic AI / ML USD 142K-196K

Agentic AI | Data Mining | Deep learning | Embeddings | Forecasting

Health and wellbeing programs | Learning and development | On-site work | Professional growth | Supportive work culture

Mid-level Full Time

Santa Clara,CA, United States

4d ago

Senior Software Engineer, AI Networking USD 152K-287K

Bash | Bayesian optimization | C++ | Data Curation | Data Curation Pipelines

Senior-level Full Time

US, CA, Santa Clara, United States

4d ago

Research Engineer, Machine Learning - Paris/London/Zurich/Warsaw EUR 50K-71K

CI/CD | CUDA | Code review | Data pipeline | Deep learning

Full health insurance | Gym subscription | Lunch vouchers | Mobility pass | Parental leave

Mid-level Full Time

Paris

5d ago

Sr. Gen AI Engineer A USD 150K-188K

AI Agents | AI Search | AWS | AWS Bedrock | Agentic Workflows

401k | Dental insurance | Medical insurance | Paid sick hours | Vision insurance

Senior-level Contract Full Time

Ridgefield Park, NJ, United States

6d ago

AI Engineer INR 2500K-4500K

A/B | A/B Testing | B testing | Computer Vision | Data Preprocessing

Senior-level Full Time

Hybrid, Hyderabad, Telangana, India R

6d ago

AI/ML Engineer, National Security, Google Public Sector USD 153K-222K

C plus plus | Data Validation | Deep learning | Feature Engineering | GPUs

Travel up to 30 percent

Senior-level Full Time

Reston, VA, USA; Washington D.C., DC, …

6d ago

Applied AI Scientist, Small Language Model and AI Training USD 218K-288K

AI Safety | Data pipeline | Distributed Training | Domain Adaptation | Fine Tuning

Donation matching | Flexible PTO | Medical coverage | Monthly lunch stipend | Team building events

Mid-level Full Time

San Francisco, California, United States

7d ago

Physical AI Engineer CNY 180K-300K

C++ | CUDA | Compilers | Isaac Sim | JAX

On-site work

Entry-level Full Time

CHN - Minhang, China

7d ago

Deep Learning Software Engineer, LLM Performance - New College Grad 2026 USD 124K-241K

C# | C++ | CUDA | Debugging | JAX

Senior-level Full Time

US, CA, Santa Clara, United States

7d ago

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