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

457 results for JAX (Skill/Tech stack)

Research Software Engineer, Multimodal AI USD 174K-253K

Agent Orchestration | Audio | C++ | Few-Shot Learning | Few-shot

Mid-level Full Time

San Jose, CA, USA

7h ago
Software Engineer III, GenAI Data Operations Research, XR GBP 225K-325K

Data Filtering | Data Processing | Data debugging | Deep learning | Distributed Computing

Senior-level Full Time

London, UK

7h ago
Senior Software Engineer, AI/ML, Geo and Gemini App USD 174K-253K

A/B | A/B Testing | B testing | C++ | Data Analysis

Senior-level Full Time

New York, NY, USA

7h ago
Senior Software Engineer, AI/ML, Ads Training USD 174K-253K

C++ | Distributed Systems | GPU | JAX | Machine Learning

Senior-level Full Time

Mountain View, CA, USA

7h ago
Research Scientist, Adaptive Compute, DeepMind USD 174K-253K

Deep learning | Generative AI | JAX | Language Models | Language Processing

Mid-level Full Time

New York, NY, USA; Mountain View, …

7h ago
Staff/Sr. ML Infrastructure Engineer, Foundation Model Compute Infra USD 175K-312K

C++ | Containers | Distributed Systems | Distributed Training | Distributed inference

Senior-level Full Time

Cupertino

12h ago
Machine Learning Engineer, Sensor Pipelines USD 175K-215K

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

Hybrid work schedule

Mid-level Full Time

Mountain View, CA, USA

18h ago
Sr. Software Engineer, AI Infrastructure USD 144K-236K

Apache Beam | Apache Flink | Apache Spark | C# | C++

Career development | Hybrid work | Stock grants

Senior-level Full Time

Sunnyvale, CA, United States

19h ago
Software Engineer, ML Platform (ML Training) USD 175K-265K

AWS | Computer Vision | DeepSpeed | Distributed Training | JAX

Mid-level Full Time

Foster City, CA

20h ago
Machine Learning Engineer - (Computer Vision) CAD 100K-149K

AWS SageMaker | Airflow | Amazon EC2 | Amazon EKS | Biometrics

Mid-level Full Time

Montreal

1d ago
Machine Learning Engineer EUR 60K-76K

ANN search | AWS SageMaker | Airflow | Amazon EC2 | Amazon EKS

Mid-level Full Time

Austria (remote) R

1d ago
Machine Learning Engineer - IV (Computer Vision) INR 1000K-1800K

ANN search | Airflow | Approximate Nearest Neighbor | CoreML | Data Ingestion

Mid-level Full Time

India (remote) R

1d ago
Physical AI Engineer (Model) KRW 31000K-31000K

C++ | CUDA | Deep learning | Imitation Learning | JAX

Senior-level Full Time

Pangyo (Software Dream Center), South Korea

1d ago
Software Technical Lead, On-Device Inference Software, Robotics, DeepMind USD 207K-301K

AI Inference | CUDA | Compilers | Computer Vision | Embedded Software

Senior-level Full Time

Mountain View, CA, USA

1d ago
Early Career Machine Learning Engineer, Applied AI USD 25K

Algorithms | Data Engineering | Data Pipelines | Data Structures | Generative AI

401k | Commuter benefits | Daily lunches | Dental insurance | Medical insurance

Mid-level Full Time

San Francisco Bay Area

1d ago
Applied AI Scientist CHF 146K-245K

C++ | Computer Vision | Data Engineering | Deep learning | Diffusion Models

Mid-level Full Time

Switzerland, Zürich, Zürich

1d ago
Research Engineer USD 180K-300K

AWS | Agent training | Automated testing | Cloud Computing | Cloud platform

Equity | Health coverage | Work with leading AI research labs

Mid-level Full Time

Mountain View

1d ago
Machine Learning Engineer (Foundation Models & Personalization) USD 140K-200K

Behavior Modeling | Causal Inference | Deep learning | Differential Privacy | Drift Detection

Own Pod product

Mid-level Full Time

San Francisco

1d ago
Senior Applied Scientist GBP 100K-120K

Black | Cloud Computing | Complex Planning | Deep learning | Git

Equity share options | Flexible office attendance | Hybrid work scheme | Remote work flexibility | Team in person days

Senior-level Full Time

London

1d ago
Staff Applied Scientist GBP 110K-160K

Black | Cloud Platforms | Data Analysis | Deep learning | GitHub

EMI share option scheme | Generous share options | Hybrid work scheme | Slack | Zoom)

Senior-level Full Time

London

1d ago
Applied AI Engineer USD 100K-150K

Ablation Studies | Computer Vision | Data Quality | Data quality monitoring | Deep learning

Senior-level Full Time

United States - Remote R

1d ago
AI Research Engineer USD 100K-150K

Ablation Studies | Accelerator hardware | Agentic Systems | Computer Vision | Data Validation

Senior-level Full Time

United States - Remote R

1d ago
AI Research Engineer USD 100K-150K

Ablation Studies | Accelerator hardware | Agentic Systems | Computer Vision | Data Quality

Senior-level Full Time

United States - Remote R

1d ago
Machine Learning Research Engineer USD 100K-150K

Ablation Studies | Accelerator hardware | Agentic Systems | Computer Vision | Data Quality

Senior-level Full Time

United States - Remote R

1d ago
Machine Learning Research Engineer USD 100K-150K

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

Senior-level Full Time

United States - Remote R

1d ago
AI/ML Software Engineer P2 (Onsite) USD 68K-131K

Data analytics | Deep learning | Distributed Computing | JAX | Machine Learning

401k match | Dental insurance | Flexible work schedules | Health insurance | Holidays

Mid-level Full Time

US-CT-EAST HARTFORD-OBG ~ 400 Main St …

1d ago
Sr. AI / ML Research Engineer-2 INR 4000K-5500K

ANNSYS Fluent | Computational Fluid Dynamics | Data Assimilation | Data Preparation | Design of Experiments

Senior-level Full Time

Bangalore IBC Knowledge Park, India

1d ago
Senior Machine Learning Engineer EUR 64K-85K

Cost Optimization | Data Pipelines | Deep learning | GPU Computing | JAX

Senior-level Full Time

Ireland

2d ago
Senior Machine Learning Engineer EUR 64K-85K

Data Preparation | GPU Computing | Inference pipeline | JAX | Machine Learning

Senior-level Full Time

Germany

2d ago
Senior Machine Learning Engineer SEK 831K-1045K

Data Preparation | Debugging | GPU Computing | Inference | JAX

Senior-level Full Time

Sweden

2d ago
Senior Machine Learning Engineer EUR 64K-85K

Cost Optimization | Data Pipelines | Data Preparation | Debugging | GPU Computing

Senior-level Full Time

Portugal

2d ago
Senior Machine Learning Engineer EUR 64K-85K

Data Preparation | Deep learning | GPU Computing | JAX | Machine Learning

Senior-level Full Time

Spain

2d ago
Senior Machine Learning Engineer PLN 257K-336K

Data Pipelines | Data Preparation | Deep learning | GPU Computing | JAX

Senior-level Full Time

Poland

2d ago
Member of Technical Staff, Machine Learning EUR 55K-79K

Data Pipelines | Fine Tuning | GPU Computing | Inference | JAX

Learning and development | Rapid iteration culture | Remote-friendly

Senior-level Full Time

Ireland

2d ago
Member of Technical Staff, Machine Learning EUR 55K-79K

Data Pipelines | Fine Tuning | GPU | JAX | Machine Learning

Senior-level Full Time

Germany

2d ago
Member of Technical Staff, Machine Learning SEK 831K-1045K

Data Pipelines | Fine Tuning | GPU Computing | Inference | JAX

Senior-level Full Time

Sweden

2d ago
Member of Technical Staff, Machine Learning EUR 55K-79K

Data Pipelines | Fine Tuning | GPU Computing | Inference | JAX

Senior-level Full Time

Portugal

2d ago
Member of Technical Staff, Machine Learning EUR 55K-79K

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

Senior-level Full Time

Spain

2d ago
Member of Technical Staff, Machine Learning PLN 280K-329K

Data Pipelines | Fine Tuning | GPU Computing | Inference | JAX

Senior-level Full Time

Poland

2d ago
Applied AI Engineer EUR 54K-75K

Agent workflows | Cost Optimization | Data Pipelines | Evaluation Frameworks | Inference

Senior-level Full Time

Ireland

2d ago
Applied AI Engineer EUR 54K-75K

Agent Orchestration | Fine Tuning | Inference | JAX | LLM Inference

Senior-level Full Time

Germany

2d ago
Applied AI Engineer SEK 775K-1030K

Agent systems | Fine Tuning | Inference | JAX | Language Models

Senior-level Full Time

Sweden

2d ago
Applied AI Engineer EUR 54K-75K

Agent workflows | Deep learning | Evaluation | Fine Tuning | Inference

Senior-level Full Time

Portugal

2d ago
Applied AI Engineer EUR 54K-75K

Agent workflows | Inference Optimization | JAX | Language Models | Language Processing

Senior-level Full Time

Spain

2d ago
Applied AI Engineer PLN 246K-321K

Data Pipelines | Fine Tuning | Inference | JAX | Language Models

Senior-level Full Time

Poland

2d ago
Robotics AI Engineer A SGD 100K-143K

Domain Randomization | Imitation Learning | IsaacGym | IsaacLab | JAX

Fast feedback cycles | High impact work on real robots | Open source contributions

Mid-level Full Time

Singapore

2d ago
Robotics Researcher, Manipulation A SGD 70K-108K

Autonomy | C++ | Contact Rich Control | Control Theory | Data collection

Competitive compensation | Equity | Open source contributions

Mid-level Full Time

Singapore

2d ago
Principal Engineer - Gen AI training and inferencing CAD 135K-191K

Deep learning | Distributed Training | Generative AI | Hugging Face | Image Generation

Hybrid work environment

Senior-level Full Time

MARKHAM, ON, Canada

2d ago
Deep Learning Researcher 1900S-2 USD 130K-160K

Bayesian optimization | Computer Vision | Data Processing | Diffusion Models | Experimental Design

401k match | Dental insurance | Employee discounts | Flexible work environment | Health insurance

Mid-level Full Time

Billerica, Massachusetts, United States

2d ago
AI Research Engineer USD 202K-312K

Blockchain | Cryptography | Decentralized networks | Distributed Systems | Federated Learning

Mid-level Full Time

Anywhere R

2d ago