399 jobs for JAX

[AI Research Div.] Research Scientist Intern - Theory (2년 이상 / 인턴)

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

Career support | Paid probation period

Entry-level Internship

Seoul

9h ago

Machine Learning Engineer (Training Optimization） CNY 50K-50K

CUDA | DeepSpeed | Distributed Training | FSDP | Gradient Checkpointing

Entry-level Full Time

Beijing, Beijing, China

9h ago

AI/ML Engineer - ATAS - Open Rank USD 124K-170K

C# | C++ | Computer Vision | Data Fusion | Data Processing

Health and welfare | Professional development | Retirement plans | Time off | Tuition reimbursement

Mid-level Full Time

Atlanta, GA

23h ago

Research Engineer, Robotics USD 184K-356K

C++ | CUDA | Computer Graphics | GPU Architectures | GPU Kernels

Senior-level Full Time

Redmond, WA

1d ago

Partner Engineer, Generative AI USD 159K-223K

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

Senior-level Full Time

Menlo Park, CA

1d ago

Research Engineer, Pretraining, DeepMind USD 174K-253K

Fine Tuning | Inference Optimization | JAX | Language Models | Large Language Models

Mid-level Full Time

New York, NY, USA

1d ago

Technical Lead Manager, TorchTPU GBP 155K-175K

C++ | Compiler optimization | Computer Vision | Distributed Computing | GPU Workloads

Senior-level Full Time

London, UK

1d ago

AI Engineer, Advanced Solutions Lab, Google Cloud USD 152K-222K

Apache Beam | Artificial Intelligence | Cloud platform | ELT | ETL

Senior-level Full Time

Chicago, IL, USA; Atlanta, GA, USA

1d ago

AI Research Engineer USD 205K-300K

Applied cryptography | Deep learning | Distributed Systems | Federated Learning | Go

Entry-level Full Time

Anywhere R

1d ago

Principal Scientist, Machine Learning - Biomolecules USD 208K-286K

AWS Batch | AWS ECS | AWS EKS | AWS S3 | AWS SageMaker

Annual incentive program | Healthcare coverage | Retirement benefits

Senior-level Full Time

Cambridge, MA USA

1d ago

Deep Learning Performance Architect CNY 152K-240K

Computer Architecture | Deep learning | Inference | JAX | Language Models

Senior-level Full Time

China, Shanghai

1d ago

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

Ablation Studies | Accelerator hardware | Agentic Systems | Computer Vision | Deep learning

Career growth | Remote work

Mid-level Full Time

United States - Remote R

1d ago

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

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

Career growth | Remote work

Mid-level Full Time

United States - Remote R

1d ago

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

C++ | CUDA | CUDNN | DeepSpeed | Docker

Mid-level Full Time

北京 R

1d ago

Audio ML Engineer II USD 160K-210K

Audio Processing | Benchmarking | Classification | DET Curves | Deep learning

401k | Commuter benefits | Company offsite | Daily in office lunch | Dental and vision

Mid-level Full Time

Global Remote R

1d ago

Staff AI Engineer (Audio) USD 185K-235K

Audio Processing | Classification metrics | Data Drift | Data analytics | Datadog

401k | Commuter benefits | Company offsite | Daily lunch | Dental/vision

Senior-level Full Time

Global Remote R

1d ago

Machine Learning Engineer GBP 80K-93K

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

Mid-level Full Time

Amsterdam, Netherlands; London, United Kingdom

2d ago

Software Engineer USD 149K-211K

Algorithms | C# | C++ | Code review | Data Analysis

Bonus | Equity | Hybrid work schedule

Mid-level Full Time

Mountain View, CA, USA R

2d ago

Software Engineer USD 149K-211K

C# | C++ | Cause analysis | Data Processing | Data Structures

Hybrid schedule

Mid-level Full Time

Sunnyvale, CA, USA R

2d ago

Research Engineer USD 147K-211K

Algorithm Design | C++ | Experimental Design | JAX | Machine Learning

Hybrid schedule

Mid-level Full Time

Mountain View, CA, USA R

2d ago

Senior Research Engineer USD 174K-252K

C plus plus | Code Reviews | Data Curation | Deep learning | JAX

Hybrid schedule

Senior-level Full Time

New York, NY, USA R

2d ago

Senior Research Engineer, On-Device Inference, Robotics, DeepMind USD 207K-301K

AI accelerators | C++ | CPU | Diffusion Models | Edge Computing

Senior-level Full Time

Mountain View, CA, USA

2d ago

Machine Learning Engineer II, Infrastructure (머신러닝 엔지니어 II - 인프라)

AWS | Apache Airflow | Apache Beam | Apache Spark | Azure

Mid-level Full Time

Seoul, Korea

2d ago

Machine Learning Engineer Perception LLM/VLM (PhD, New Grad) USD 170K-216K

Continual Learning | Data Processing | Deep learning | JAX | Language Models

Bonus program | Company benefits program | Equity incentive plan

Entry-level Full Time

Mountain View, CA USA; San Francisco, …

2d ago

Staff Machine Learning Engineer, Multi-Modal Perception USD 251K-310K

C plus plus | Computer Vision | Data Analysis | Deep learning | JAX

Senior-level Full Time

Mountain View, CA USA; San Francisco, …

2d ago

Senior Machine Learning Engineer, End‑to‑End Autonomous Driving USD 184K-356K

Active Learning | Automated Data Labeling | CI/CD | Code review | Curriculum learning

Senior-level Full Time

US, CA, Santa Clara, United States

2d ago

Matterport – Senior Machine Learning/Computer Vision Engineer – 3D Reconstruction and Semantic Understanding USD 190K-253K

3D Scene | 3D Scene Reconstruction | Agile | Appearance modeling | CUDA

401k matching | Commuter benefits | Comprehensive healthcare coverage | Dental coverage | Employee resource groups

Senior-level Full Time

Sunnyvale, United States

2d ago

Machine Learning Research Engineer | Kilby Labs USD 149K-258K

C++ | DPO | Deep learning | Embeddings | Few-Shot Learning

Mid-level Full Time

United States

2d ago

AI and Data Scientist HUF 14400K-22882K

AWS | Anomaly Detection | Artificial Intelligence | Azure | Computer Vision

Collaborative workspaces | Flexible working arrangements | Learning platform access | LinkedIn Learning access

Mid-level Full Time

Budapest, PE, Hungary

3d ago

Mid-Level Machine Learning Engineer USD 110K-200K

ASIC | C# | C++ | CoreML | Edge AI

Career development support | Mentorship | Open source contribution opportunities

Mid-level Full Time

Fremont , CA

3d ago

Senior Machine Learning Engineer USD 200K-280K

ARM Ethos | ASIC | C plus plus | DSP | Edge AI

Senior-level Full Time

Fremont , CA

3d ago

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

Accelerators | Computer Vision | Data Quality | Data labeling | Data quality monitoring

Remote work

Mid-level Full Time

United States - Remote R

3d ago

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

Accelerator hardware | Computer Vision | Data Quality | Data quality monitoring | Deep learning

Remote work

Mid-level Full Time

United States - Remote R

3d ago

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

Accelerator hardware | Computer Vision | Data labeling | Deep learning | Distributed Training

Mid-level Full Time

United States - Remote R

3d ago

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

Accelerators | Computer Vision | Data Modeling | Data Quality | Data Validation

Career growth | Remote work

Mid-level Full Time

United States - Remote R

3d ago

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

Ablation Studies | Accelerator hardware | Agentic Systems | Data Quality | Deep learning

Mid-level Full Time

United States - Remote R

3d ago

Postdoctoral Position (m/f/d) Machine Learning for the Acceleration of Simulations EUR 48K-55K

Computational Fluid Dynamics | Finite element | Finite element method | Fluid Dynamics | High Performance

Flexible working hours | International research environment

Entry-level Full Time

Kiel, Schleswig-Holstein, DE, 24105

3d ago

Scientist, Structural Chemistry AI/ML USD 146K-189K

ADMET modeling | Applicability Domain | Cheminformatics | Cheminformatics toolkits | Deep learning

Dental insurance | Health insurance | Life insurance | Paid time off | Vision insurance

Entry-level Full Time

US - CA - Foster City, …

3d ago

Sr Gen AI Engineer USD 87K-140K

AI Safety | AKS | ARM Templates | Active Directory | Azure

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

Senior-level Full Time

Dallas, United States

3d ago

Senior Scientist, Machine Learning (Biologics Design) USD 169K-219K

Active Learning | Amber | Antibody engineering | Bayesian optimization | CHARMM

Senior-level Full Time

US - CA - Foster City, …

3d ago

Lead Applied Data Scientist - Digital Item RecSys (applied ML, deep learning) USD 132K-286K

A/B | A/B Testing | B testing | Deep learning | Experimentation

401k | Employee discount | Health benefits | Long-term disability | Paid national holidays

Senior-level Full Time

7000 Target Pkwy N,NCD-0375 Brooklyn Park,MN …

3d ago

Technical Leader: AI Systems Architecture CHF 128K-150K

Data Preprocessing | Diffusion Models | Distributed Training | Inference Optimization | JAX

Cross-functional collaboration

Senior-level Full Time

Crissier, Switzerland

4d ago

Inference Intern USD 60K-142K

C++ | Collective communication | Compilers | Consensus Protocols | Consistency models

Daily meals | Direct mentorship | Housing support | Paid internship

Entry-level Internship

San Jose

4d ago

Machine Learning Scientist, BioML USD 200K-330K

AWS | Azure | Bioinformatics | Cloud Computing | Computational Biology

401k employer match | Equity participation | Health, dental, vision insurance | Paid time off | Professional development

Mid-level Full Time

Emeryville, California, United States; Hybrid (2-3 … R

5d ago

Senior Research Engineer, Olmo + Molmo USD 146K-220K

Agentic Systems | Amazon Web Services | Cloud Computing | Cloud platform | Containerization

Family leave | Paid sick leave | Paid vacation

Senior-level Full Time

Seattle, WA

5d ago

AI Engineer USD 152K-200K

API Development | Bias Mitigation | Cloud Computing | Data Pipelines | Data Preparation

Senior-level Full Time

Lincoln, NE, US

5d ago

Sr. Physical AI Research Scientist CAD 140K-180K

AI alignment | Artificial Intelligence | Computer Vision | Constitutional AI | Continual Learning

Hybrid work schedule

Senior-level Full Time

Toronto, ON, CA

6d ago

Principal ML Engineer USD 250K-350K

ASR | Audio augmentation | CNN | CTC | DSP

Senior-level Full Time

Palo Alto, CA

6d ago

Research Scientist / Research Engineer A INR 1600K-2400K

Data collection | Data shaping | Fine Tuning | Gradient Free Optimization | Human Feedback

Annual travel stipend | Flexible work arrangements | Global remote work | Meal Delivery Stipend | Medical coverage

Senior-level Full Time

India

6d ago

Research Scientist / Research Engineer A CAD 140K-179K

Data Engineering | Fine Tuning | Human Feedback | JAX | Learning from Human Feedback

Meal Delivery Stipend | Medical coverage | Paid time off | Remote work | Team offsites

Senior-level Full Time

Canada

6d ago

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