403 jobs for JAX

Software Engineer, Research Acceleration USD 350K-475K

Distributed Systems | Evaluation Frameworks | Experiment tracking | JAX | Machine Learning

Dental insurance | Health insurance | Paid parental leave | Relocation support | Unlimited PTO

Senior-level Full Time

San Francisco

9h ago

ML/AI Engineer - Vehicle Intelligence USD 189K-261K

Data Privacy | Digital Twins | Digital modeling | Edge Computing | Foundation Models

Mid-level Full Time

Sunnyvale, United States

14h ago

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

C++ | CUDA | CUDNN | Containerization | DeepSpeed

Mid-level Full Time

北京 R

19h ago

Physical AI Engineer - SW USD 144K-180K

Agentic Workflows | CFD | Data Pipelines | Differentiable Optimization | Domain Randomization

Senior-level Full Time

San Jose, California, United States

21h ago

Senior Computer Vision Engineer A TRY 840K-1080K

Adversarial Networks | Autoencoders | C# | C++ | Computer Vision

Collaborative mission driven culture | Hybrid/remote work options | Premium cloud access | Research-focused work

Senior-level Full Time

Ankara, Türkiye

1d ago

Machine Learning Research Engineer USD 161K-189K

AWS | Azure | Bias Variance | Bias-Variance Tradeoff | C++

Flexible hybrid work model | Mental health counseling | Mentorship programs | Paid parental leave | Paid volunteer time off

Mid-level Full Time

New York, US, New York

1d ago

Intern in the CAVES & PANGAEA Team, Machine Learning Development EUR 25K-27K

Convolutional Neural Networks | Data Fusion | Deep learning | JAX | Keras

Entry-level Internship

Porz-Wahn, DE

1d ago

Foundation Model Data, Software Engineer USD 213K-263K

Apache Beam | Apache Flume | C++ | CI/CD | Distributed Systems

401k match | Baby bonding leave | Dental insurance | Disability insurance | Health insurance

Senior-level Full Time

Mountain View, CA, USA; San Francisco, …

1d ago

Staff ML Engineer, Generative Model Performance & Efficiency USD 251K-310K

Data parallelism | Diffusion Models | Efficient Attention | Expert parallelism | Flax

Senior-level Full Time

Mountain View, California, United States, New …

1d ago

Co-Op, LS AI, ML Scientist for Protein Engineering USD 162K-288K

Active Learning | Antibody engineering | Biophysics | Data Analysis | Developability

Collaborative cross-disciplinary environment | Hands-on research experience

Entry-level Full Time

San Francisco, CA USA

1d ago

Machine Learning Scientist CAD 100K-150K

Airflow | Amazon SageMaker | Anomaly Detection | Computer Vision | Data Preprocessing

Climbing wall | Gym | On-site work | Squash court | Steam room

Mid-level Full Time

North Vancouver, British Columbia

2d ago

Staff Software Engineer, AI/ML, YouTube Ads USD 207K-301K

A/B | A/B Testing | B testing | Data Structures | Data structures algorithms

Senior-level Full Time

Mountain View, CA, USA

2d ago

Chercheur·euse en deep learning / computer vision H/F A EUR 70K-80K

Computer Vision | Deep learning | Flax | Git | JAX

BSPCE stock options | Remote work policy

Senior-level Full Time

Paris, FR - Remote/Hybrid R

2d ago

Staff Applied Scientist GBP 110K-160K

Cloud Computing | Complex Planning | Data Analysis | Deep learning | Explainable AI

EMI share option scheme | Generous share options | Hybrid work schedule | Mentoring and coaching culture

Senior-level Full Time

London

2d ago

Senior Applied Scientist GBP 100K-120K

Black | Cloud Computing | Deep learning | Git | GitHub

EMI share option scheme | Remote work hybrid scheme | Share options | Tax efficient equity scheme | Team days in person

Senior-level Full Time

London

2d ago

AI Research Engineer USD 100K-150K

Ablation Studies | Accelerator hardware | Data Quality | Data labeling | Data quality monitoring

100 percent remote | Career growth | Full-time employment | W2 employment

Mid-level Full Time

United States - Remote R

2d ago

AI Research Engineer USD 100K-150K

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

Mid-level Full Time

United States - Remote R

2d ago

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

Ansible | Autogen | Bash | CI/CD | CloudWatch

Child education allowance | Employee stock purchase scheme | Health reimbursement | Life insurance | Medical insurance

Senior-level Full Time

Riyadh, Saudi Arabia

2d ago

Senior Software Engineer, ML INR 3700K-5000K

Algorithms | Deep learning | GPU Performance | GPU Performance Optimization | Git

Senior-level Full Time

India, Bengaluru

2d ago

Software Dev Engineer II, Stores Foundational AI -SFAI USD 165K-223K

Async Rollouts | Batching | C++ | CUDA | Cluster computing

401k matching | Adoption reimbursement | Dental insurance | Employee assistance program | Flexible spending accounts

Mid-level Full Time

Palo Alto, California, USA

2d ago

Software Dev Engineer II, Stores Foundational AI -SFAI USD 143K-194K

CUDA | Data Pipelines | Distributed Training | Dynamo | Experiment tracking

401k matching | Employee assistance program | Health insurance | Paid time off | Parental leave

Mid-level Full Time

Seattle, Washington, USA

2d ago

Sr Software Dev Engineer, Stores Foundational AI -SFAI USD 168K-227K

Async Rollouts | Batching | Distributed Systems | Experiment tracking | GPU Utilization

401k matching | Dental insurance | EAP | Flexible spending accounts | Health insurance

Senior-level Full Time

Seattle, Washington, USA

2d ago

AI Research Engineer USD 100K-150K

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

Career growth | Mentorship | Remote work

Mid-level Full Time

United States - Remote R

2d ago

AI Research Engineer USD 100K

Ablation Studies | Accelerator hardware | Computer Vision | Data Quality | Data quality monitoring

Equal opportunity employer | Health benefits | Professional growth | Remote work

Mid-level Full Time

United States - Remote R

2d ago

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

Agentic Systems | Computer Vision | Deep learning | Distributed Training | JAX

Mid-level Full Time

United States - Remote R

2d ago

AI Research Engineer USD 100K-150K

Ablation Studies | Agentic Systems | Computer Vision | Data Quality | Data quality monitoring

Benefits | Career growth | H1B transfer support | Remote work

Mid-level Full Time

United States - Remote R

2d ago

AI Research Engineer USD 100K-150K

Computer Vision | Data Quality | Data quality monitoring | Deep learning | Distributed Training

Benefits | Remote work

Mid-level Full Time

United States - Remote R

2d ago

Robotics Researcher, Perception & Vision A SGD 100K-214K

3D Geometry | 3D Scene | 3D Scene Reconstruction | C++ | Computer Vision

Equity | Open source contributions | Real-world robotics deployment

Senior-level Full Time

Singapore

3d ago

Robotics Researcher, Locomotion A SGD 60K-108K

Body Dynamics | Isaac-Gym | JAX | Model-Predictive Control | Mujoco

Equity | Health insurance | Learning opportunities | Open Source contribution | Paid time off

Mid-level Full Time

Singapore

3d ago

Robotics Researcher, Manipulation A SGD 64K-108K

C++ | Contact Rich Control | Control Theory | Data collection | Grasp Planning

Open source contributions | Work on Physical Robots

Mid-level Full Time

Singapore

3d ago

AI Research Engineer USD 205K-344K

Applied cryptography | Blockchain | Distributed Systems | Federated Learning | Fine Tuning

Entry-level Full Time

Anywhere R

3d ago

AI Research Scientist USD 240K-350K

Benchmarking | Convolutional Neural Networks | Diffusion Models | Distributed Training | Federated Learning

401k match | Continuing education support | Equity options | Flexible time off | Free parking

Mid-level Full Time

Austin, TX

3d ago

Research Scientist - Distributed Machine Learning USD 180K-287K

BF16 | CUDA | CUDA kernels | DeepSpeed | Distributed Training

401k | Dental insurance | Disability insurance | Employee assistance program | Health insurance

Mid-level Full Time

Sunnyvale, CA

3d ago

Machine Learning Infrastructure Engineer USD 216K-330K

CUDA | DeepSpeed | Distributed Systems | Distributed Training | FSDP

Mid-level Full Time

Sunnyvale, CA

3d ago

High Performance Computing Software Engineer - Supercomputing AED 254K-306K

Containerization | Debugging | DeepSpeed | Distributed Systems | GPU Kernel

Mid-level Full Time

Abu Dhabi

3d ago

Machine Learning Engineer HKD 321K-321K

C++ | CUDA | CUDNN | Cloud Computing | Distributed Training

Mid-level Full Time

Hong Kong, Hong Kong; Sydney, Australia

4d ago

Senior Machine Learning Engineer USD 180K-250K

Computer Vision | Data Pipelines | Data labeling | Deep learning | Embedding Models

100 percent remote | 13 paid holidays | 401k plan | Dental insurance | Medical insurance

Senior-level Full Time

Remote USA R

4d ago

Staff Machine Learning Engineer USD 278K-330K

Automatic Speech Recognition | Cloud Computing | Data Augmentation | Data Pipelines | Data Preprocessing

Senior-level Full Time

Mountain View, CA

4d ago

Senior Machine Learning Engineer USD 230K-265K

Automatic Speech Recognition | Cloud Computing | Data Augmentation | Data Preprocessing | Decoding strategies

Senior-level Full Time

Mountain View, CA

4d ago

Senior Machine Learning Engineer USD 208K-263K

Accelerated inference | Active Learning | BEV | Birds Eye View | C++

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

Senior-level Full Time

San Francisco, CA

4d ago

Staff Machine Learning Engineer USD 273K-345K

Active Learning | CUDA | Cameras | Data Engineering | Data Mining

401k | Disability insurance | Equity | Flexible spending account | Flexible time off

Senior-level Full Time

San Francisco, CA

4d ago

Senior Machine Learning Engineer CAD 129K-180K

AWS | Airflow | Computer Vision | Data Ingestion | Face detection

Senior-level Full Time

Montreal

4d ago

Senior Machine Learning Engineer USD 70K-300K

C++ | Convolutional Neural Networks | JAX | Model Drift | Model Evaluation

Senior-level Full Time

Irvine, CA

4d ago

Staff AI Inference and Acceleration Engineer USD 180K-275K

AI Inference | Benchmarking | C++ | CUDA | Computer Architecture

Senior-level Full Time

San Jose, CA

4d ago

Research Engineer / Research Scientist, Vision USD 350K-850K

Agentic Systems | Benchmarking | Computer Vision | Data Pipelines | Deep learning

Flexible working hours | Generous vacation | Hybrid work option | Optional equity donation matching | Parental leave

Senior-level Full Time

New York City, NY; San Francisco, …

4d ago

Research Scientist, Recommendation Systems, DeepMind USD 174K-253K

Deep learning | Experimentation | JAX | Language Models | Language Processing

Mid-level Full Time

Mountain View, CA, USA; New York, …

5d ago

Principal Engineer, Machine Learning INR 4000K-5000K

Artificial Intelligence | Computer Vision | Deep learning | Docker | Foundation Models

Senior-level Full Time

Remote, India R

5d ago

Associate Principal Engineer, Machine Learning INR 1800K-3300K

Adversarial Networks | Agent Orchestration | Agent systems | Computer Vision | Deep learning

Mid-level Full Time

Remote, India R

5d ago

Applied Scientist / Research Engineer KRW 44000K-50000K

Agent systems | Data Curation | Distributed Training | GPU Computing | JAX

Food allowance | Gym membership allowance | Health insurance | Transportation reimbursement

Entry-level Full Time

Seoul

5d ago

Senior/Staff Applied Scientist/Research Engineer

Agents | CI/CD | Data Curation | Data Pipelines | Deep learning

Senior-level Full Time

Seoul

5d ago

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