216 jobs for JAX

校招-Ai研究科学家-大语言模型/视觉语言模型算法与后训练（博士优先） CNY 500K-500K

Adapters | Direct Preference Optimization | Fine Tuning | Flax | Function design

None Full Time

上海

12h ago

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

AI Search | API Development | AWS | AWS Bedrock | Azure

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

Senior-level Contract Full Time

Ridgefield Park, NJ, United States

14h ago

Robotics Manipulation Engineer USD 170K-240K

Adaptive Control | Automation | C++ | Deep learning | Gazebo

Senior-level Full Time

Fremont, CA

18h ago

Partner Engineer, Generative AI INR 2500K-3000K

AWS | Azure | C plus plus | C# | Cloud Computing

Senior-level Full Time

Bangalore, India | Mumbai, India

18h ago

Staff Software Engineer, Torch TPU USD 207K-300K

CUDA | Computer Vision | Data Processing | Debugging | Distributed Systems

Senior-level Full Time

Sunnyvale, CA, USA

18h ago

AI Software Engineer - Neuromorphic Computing EUR 43K-60K

Benchmarking | C++ | CI/CD | CUDA | Continuous integration

Documentation and code review opportunities | Fully home based | Occasional onsite visits

Mid-level Full Time

Virtual - NLD NL, Netherlands R

1d ago

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

AI Safety | AI Search | AWS SageMaker | Azure AI | Azure AI Search

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

Senior-level Contract Full Time

Ridgefield Park, NJ, United States

1d ago

Software Engineer, ML (Training and Inference) GBP 70K-90K

Accelerators | Data Structures | Data structures algorithms | Data version control | Distributed Computing

Hybrid working

Mid-level Full Time

London

1d ago

Solution Architect Graduate (BytePlus) - 2026 Start (BS/MS) - Vietnam USD 136K-250K

Caching | Context engineering | Docker | FastAPI | Generative AI

Entry-level Full Time

Ho Chi Minh City, Ho Chi …

1d ago

Staff Software Engineer, AI/ML Data Processing USD 207K-300K

APIs | Checkpointing | Cloud technologies | Data Processing | Debugging

Senior-level Full Time

Sunnyvale, CA, USA

1d ago

AI Research Engineer USD 300K-425K

Applied cryptography | Blockchain Protocols | Cryptography | Distributed Systems | Federated Learning

Entry-level Full Time

Anywhere R

1d ago

Machine Learning Engineer, Geometry Team USD 175K-215K

3D Spatial Reasoning | C++ | Computer Vision | Distributed Training | JAX

Senior-level Full Time

Kirkland, Washington, United States

2d ago

AI Engineer, AVP INR 1500K-6000K

AWS | Adapters | Agent Development Kit | Agent systems | Azure

Accident and term life insurance | Best in class leave policy | Childcare assistance reimbursement | Comprehensive hospitalization insurance | Employee assistance program

Executive-level Full Time

Pune - Margarpatta, India

2d ago

Research Engineer USD 120K-238K

AITemplate | Architecture Search | Attention Optimization | Computer Vision | CoreML

Senior-level Full Time

Seattle, United States

2d ago

Data Scientist (F/M) EUR 14K-18K

Audience Segmentation | Azure | DBT | Databricks | Generative AI

Career coaching | Company concierge service | Employee shop | Mobility opportunities | Partially remote work

Mid-level Full Time

Issy-les-Moulineaux, FR, 92130

2d ago

Sr. Gen AI Engineer A USD 156K-219K

AI Search | AWS | Amazon SageMaker | Azure AI | Azure AI Search

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

Senior-level Contract Full Time

Ridgefield Park, NJ, United States

2d ago

Principal Staff Software Engineer, AI Training Platform USD 207K-340K

Automatic mixed precision | C++ | Container Orchestration | Containerization | Data parallelism

Commute to office based on team needs | Hybrid work options | Mentorship and career growth | Work from home

Executive-level Full Time

Mountain View, CA, United States

2d ago

Sr. Staff Software Engineer, AI Infra USD 198K-326K

C++ | CUDA | DeepSpeed | Distributed Training | GNN

Senior-level Full Time

Mountain View, CA, United States

2d ago

Research Scientist, Deep Learning USD 174K-252K

Debugging | Deep learning | Distributed Training | JAX | Learning Theory

Senior-level Full Time

Mountain View, California, US; San Francisco, …

2d ago

AI Research Scientist | Research & Development USD 200K-300K

C++ | CUDA | Distributed Training | GPU Optimization | High Performance

Dental insurance | Dependent care options | Employer-paid life insurance | Group term life insurance | HSA & FSA options

Mid-level Full Time

New York, London

2d ago

Machine Learning Engineer USD 138K-250K

A/B | A/B Testing | B testing | Cost Optimization | Deep learning

401k | AD D Insurance | Dental and vision insurance | Extended maternity leave | Extended paternity leave

Senior-level Full Time

USA, Palo Alto

2d ago

Systems Engineering Intern (Machine Learning) USD 62K-124K

Agentic AI | C# | C++ | DPO | Deep learning

Entry-level Full Time Internship

Dallas, TX, United States

2d ago

Research Scientist Intern, Robot Foundation Models (PhD) USD 108K-200K

Behavioral models | Control Theory | Dynamics | Imitation Learning | JAX

Entry-level Internship

Redmond, WA | Burlingame, CA

2d ago

Software Engineer, Generative AI Media, Vertex AI USD 147K-211K

C++ | Data Analysis | JAX | Java | Machine Learning

Benefits | Bonus | Equity

Mid-level Full Time

Sunnyvale, CA, USA

2d ago

Associate Process Manager INR 2500K-3199K

AWS | Azure | CI/CD | Docker | Fine Tuning

Mid-level Full Time

Pune, Maharashtra, India

2d ago

2026 Graduate Machine Learning Engineer - Applied AI GBP 30K-34K

Benchmarking | C++ | CUDA | Cloud Platforms | Deep learning

Annual leave policy | Dental plan | Employee assistance programme | Flexible working | Health cash plan

Entry-level Full Time

Bristol, UK

2d ago

Software Engineer - Pytorch GBP 33K-50K

C++ | Code Quality | Code review | Compiler development | Integration Testing

Barista service | Dental plan | Employee assistance programme | Flexible working | Generous annual leave

Senior-level Full Time

Bristol, UK; Cambridge, UK

2d ago

Senior Applied Scientist GBP 230K-340K

Black | Cloud Computing | Complex Planning | Data Analysis | Data Processing

Hybrid work | Mentoring and coaching | Share options | Slack | Zoom)

Senior-level Full Time

London

2d ago

Machine Learning Engineer (NYC) USD 110K-250K

Autoencoders | Data Pipelines | Deep learning | Diffusion Models | Distributed Computing

Dental insurance | Equity | Health insurance | Vision insurance

Mid-level Full Time

New York HQ 🗽

2d ago

Machine Learning Engineer INR 2500K-3000K

Deep learning | Evaluation Frameworks | JAX | Kubernetes | Language Models

Senior-level Full Time

India - Hyderabad

3d ago

Machine Learning Engineer INR 2500K-3000K

CI/CD | Data Filtering | Deep learning | Evaluation Frameworks | Explainability tools

Senior-level Full Time

India - Hyderabad

3d ago

Senior Software Engineer, AI Storage USD 184K-287K

Algorithms | Bash | C# | C++ | CUDA

Senior-level Full Time

US, CA, Santa Clara, United States

3d ago

Data Science Intern USD 46K-103K

Data Analysis | Deep learning | Foundation Models | JAX | Machine Learning

Medical benefits | Retirement plan | Sick time

Entry-level Full Time Internship

US338 MA Cambridge - 320 Bent …

3d ago

Software Engineer (MESA), Center for Computational Astrophysics USD 150K-190K

C# | C++ | Code profiling | Code review | Documentation

Mid-level Full Time

162 5th Avenue, New York, NY …

3d ago

Senior ML Engineer – Distributed RL & Post-Training Infrastructure A USD 204K-350K

Automated testing | Cryptography | Direct Preference Optimization | Distributed Systems | Docker

Senior-level Full Time

Remote R

3d ago

Research Scientist - Decentralized LLM Pre-Training A USD 200K-300K

Distributed Training | Federated Learning | JAX | Language Models | Large Language Models

Mid-level Full Time

Remote R

3d ago

Machine Learning Research Intern (Summer 2026) USD 80K-120K

Deep learning | Diffusion Models | GPU Kernels | JAX | Model Optimization

Access to cutting-edge compute | Daily lunch

Entry-level Internship

South San Francisco

3d ago

2026 Intern, Agentic AI Solutions (Summer/Fall) USD 88K-126K

Algorithms | Data Structures | Deep learning | Edge Computing | Hugging Face

Flexible work hours | Paid internship

Entry-level Internship

665 Clyde Avenue, Mountain View, CA, …

3d ago

Master Thesis Deep Learning for Multi-Channel Vision in Product Manufacturing EUR 30K-30K

Computer Vision | Deep learning | JAX | Machine Learning | PyTorch

Collaborative environment | Professional development opportunities | Remote work option

Entry-level Full Time

Renningen, BW, Germany

3d ago

AI Scientist - Palo Alto USD 250K-340K

Distributed Systems | Distributed Training | JAX | Kubernetes | Language Models

Coaching | Equity | Gym reimbursement | Health insurance | Meal stipend

Mid-level Full Time

Palo Alto

4d ago

Machine Learning / Senior Machine Learning Scientist / Engineer USD 170K-233K

Deep learning | Generative Models | JAX | PyTorch | Python

Bonuses | Onsite work

Senior-level Full Time

South San Francisco, CA

4d ago

Data Scientist USD 125K-160K

AWS | Azure | Bash | Data Visualization | DevSecOps

Flexible work hours | Professional development opportunities

Senior-level Full Time

McLean, VA, United States

4d ago

2026 Future Talent Program - AI/ML - Intern USD 39K-105K

AWS | Deep learning | GPU | Graph Neural Networks | JAX

Collaborative environment | Internship | Training opportunities

Entry-level Full Time Internship

USA - Massachusetts - Cambridge (320 … R

4d ago

Staff Machine Learning Engineer, Adobe GenStudio USD 172K-306K

AI systems | Algorithm Evaluation | CUDA | Data Engineering | Data Management

Flexible schedule | Health insurance | Professional development support | Retirement plan

Senior-level Full Time

San Jose, United States

4d ago

Principal Software Engineer, TensorRT-LLM USD 272K-431K

AI Inference | AI inference frameworks | C# | C++ | Deep learning

Benefits | Equity

Senior-level Full Time

US, CA, Santa Clara, United States

4d ago

AI/ML Specialist Solution Architect (Early Talent) USD 180K-225K

Cloud infrastructure | Docker | Git | JAX | Kubernetes

Mentorship | Professional development | Remote work | Team environment

Senior-level Full Time

United States

4d ago

Software Engineer III, AI/ML GenAI, GCP, Performance USD 147K-211K

C++ | Distributed Systems | JAX | Java | Language Processing

Benefits

Senior-level Full Time

Sunnyvale, CA, USA

4d ago

AI Scientist - Audio EUR 60K-78K

Distributed Systems | JAX | Language Models | Large Language Models | MLOps

Food vouchers | Gym membership reimbursement | Health insurance | Parental leave | Transportation allowance

Mid-level Full Time

Paris

5d ago

Applied Research Intern USD 100K-120K

AI | Benchmarking | Data strategy | Deep learning | JAX

Career growth opportunities | Flexible work location | Hybrid work model | Impactful work environment

Entry-level Internship

San Francisco Bay Area

5d ago

Applied Research Engineer, Agents USD 250K-300K

AI research | Autonomous Agents | Benchmarking | Data Pipelines | Data Science

Growth opportunities | Hybrid work environment | Impact focus

Mid-level Full Time

San Francisco Bay Area

5d ago

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