495 jobs for JAX

Engineering Manager, GenAI, Google Maps Immersive Navigation USD 207K-300K

C# | C++ | Code review | Computer Vision | Data Processing

Senior-level Full Time

Mountain View, CA, USA

1d ago

Staff ML Engineer, Perception Research USD 251K-310K

3D Perception | Autoregressive models | Data Pipelines | Data parallelism | Diffusion Models

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

Senior-level Full Time

Mountain View, CA, USA ; New … R

2d ago

Staff/Lead Machine Learning Engineer, Behavior & Planning USD 235K-352K

C++ | Diffusion Models | Foundation Models | Generative Modeling | Imitation Learning

Senior-level Full Time

Mountain View, California (HQ)

2d ago

Multimodal Generative AI Research Scientist (Intelligent Creation) – Global Frontier Tech Recruitment Program - 2027 Start (PhD) USD 136K-250K

DPO | Efficient Model Design | Fine Tuning | GRPO | Generative AI

Entry-level Full Time

Seattle, Washington, United States

2d ago

Research Engineer, Gemini Retrieval, DeepMind USD 174K-252K

C plus plus | Information Retrieval | JAX | Language Models | Language Processing

Mid-level Full Time

Mountain View, CA, USA; San Francisco, …

2d ago

AI Inference Tools Engineer USD 167K-242K

Agent Frameworks | Benchmarking | C++ | CUDA | Dashboards

401k matching | Flexible paid time off | Health insurance | Hybrid work options | Local meetups

Senior-level Full Time

United States, United Kingdom R

2d ago

Senior Applied Scientist GBP 92K-130K

Cloud | Complex Planning | Data Analysis | Dataset Processing | Deep learning

Hybrid work option | Learning and development | Remote work flexibility | Stock options | Team meetups

Senior-level Full Time

London

2d ago

Staff Applied Scientist GBP 150K-190K

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

Hybrid work schedule | Mentoring and coaching | Slack | Zoom)

Senior-level Full Time

London

2d ago

Senior Data Scientist USD 136K-237K

Amazon Athena | Amazon SageMaker | Anomaly Detection | Apache Iceberg | Apache Spark

Always on recognition | Annual wellness and community outreach days | Comprehensive benefits | Flexible work environment | Global collaboration and networking opportunities

Senior-level Full Time

Cordoba, Argentina

3d ago

Applied Scientist, Safe RL, Robotics, SAF Lab USD 142K-193K

Barrier functions | Constrained Markov Decision Process | Control barrier functions | Distributed Training | Domain Randomization

401k plan | Maternity and parental leave options | Medical, dental, and vision coverage | Paid time off (PTO)

Mid-level Full Time

Pasadena, California, USA

3d ago

ML Software Research Engineer USD 143K-264K

C++ | Data Preprocessing | Device Machine Learning | Foundation Models | Go

Mid-level Full Time

Pittsburgh

3d ago

Senior Machine Learning Engineer, Perception USD 184K-287K

3D Geometry | BEV Representation | Camera | Computer Vision | Deep learning

401k plan | Free lunch | Salary and benefits package | Unlimited beverages | Unlimited snacks

Senior-level Full Time

Santa Clara, CA

3d ago

LLM Post-training Engineer Intern (Research & Product) - 2027 Summer USD 119K-180K

Content Classification | DPO | Data Generation | Data Processing | DeepSpeed

Development workshops | Hands-on experience | Industry exposure | Learning programs | Social events

Entry-level Internship

San Jose, California, United States

3d ago

Data Scientist GBP 70K-80K

3D Computer Vision | Bayesian Nonparametric | Computer Vision | Deep learning | Diffusion Models

Annual leave | Bike to work scheme | Childcare support | EV salary sacrifice | Employee assistance programme

Entry-level Full Time

London, United Kingdom R

3d ago

Principal Data Scientist GBP 80K-100K

3D Computer Vision | Bayesian Non Parametric | CUDA | Cloud processing | Computer Vision

Annual leave | Bike2work scheme | EV salary sacrifice | Employee assistance programme | Employer pension contribution

Senior-level Full Time

London, United Kingdom R

3d ago

Senior ML Engineer (AI Research, Physical AI) A RON 234K-313K

CI/CD | Computer Vision | Deep learning | Distributed Checkpointing | Distributed Training

Autonomy and ownership | Career growth opportunities | Continuous learning support | Flexible work environment | Innovative culture

Senior-level Full Time

Romania

3d ago

Senior ML Engineer (AI Research, Physical AI) A EUR 78K-120K

CI/CD | Computer Vision | Control Systems | Deep learning | Distributed Checkpointing

Autonomy and ownership | Career growth opportunities | Continuous learning support | Experimentation focused culture | Flexible work environment

Senior-level Full Time

Italy

3d ago

Senior ML Engineer (AI Research, Physical AI) A EUR 78K-108K

Actor-critic | CI/CD | Computer Vision | Deep learning | Distributed Checkpointing

Autonomy and ownership | Career growth opportunities | Continuous learning support | Flexible work environment | Impactful AI research projects

Senior-level Full Time

Portugal

3d ago

Senior ML Engineer (AI Research, Physical AI) A EUR 78K-108K

CI/CD | Computer Vision | Control Systems | Deep learning | Distributed Training

Autonomy and ownership | Career growth opportunities | Continuous learning support | Flexible work environment | Innovative culture

Senior-level Full Time

Netherlands

3d ago

Senior ML Engineer (AI Research, Physical AI) A EUR 75K-108K

Action models | CI/CD | Code Reviews | Computer Vision | Data Quality

Autonomy and ownership | Career growth opportunities | Continuous learning support | Flexible work environment | Innovative research culture

Senior-level Full Time

Ireland

3d ago

Senior ML Engineer (AI Research, Physical AI) A CHF 129K-192K

Computer Vision | Continuous Deployment | Continuous integration | Continuous integration Continuous deployment | Control Systems

Autonomy and ownership | Career growth opportunities | Continuous learning support | Flexible work environment | Innovation focused experimentation

Senior-level Full Time

Switzerland

3d ago

Senior ML Engineer (AI Research, Physical AI) A EUR 75K-108K

CI/CD | Code review | Computer Vision | Continuous integration | Control Systems

Autonomy and ownership | Career growth opportunities | Continuous learning support | Flexible work environment | Influence on AI physical systems development

Senior-level Full Time

France

3d ago

Senior ML Engineer (AI Research, Physical AI) A EUR 78K-108K

Action Trajectory Optimization | CI/CD | Code review | Computer Vision | Data Quality

Autonomy and ownership | Career growth opportunities | Continuous learning support | Flexible work environment | International collaboration

Senior-level Full Time

Germany

3d ago

Senior ML Engineer (AI Research, Physical AI) A EUR 78K-108K

CI/CD | Computer Vision | Control Systems | Deep learning | Distributed Training

Autonomy and ownership | Career growth opportunities | Continuous learning support | Flexible work environment | International collaboration

Senior-level Full Time

Spain

3d ago

Lead Machine Learning Engineer USD 156K-235K

A/B | A/B Testing | Airflow | Apache Beam | Apache Spark

401k | Disability benefits | Life insurance | Medical/Dental/Vision | PTO

Senior-level Full Time

New York, NY, US, 10036

3d ago

Internship - Embodied AI & Humanoid Robotics USD 80K-116K

Computer Vision | Deep learning | Foundation Models | Imitation Learning | Isaac Sim

Covered travel | Health insurance coverage | Immigration support | Monthly transit card | Professional development

Entry-level Internship

Cambridge, MA

3d ago

ML Engineer, Foundation Model Infrastructure USD 175K-215K

Bayesian Inference | C++ | CI/CD | Distributed Systems | Experiment tracking

Senior-level Full Time

Mountain View, CA, USA; San Francisco, …

3d ago

Staff Machine Learning Engineer – VLM/LLM Evaluation USD 238K-302K

C++ | Data Pipelines | Deep learning | Distributed Systems | Evaluation Frameworks

401k match | Dental insurance | Disability insurance | Health insurance | Holidays

Senior-level Full Time

Mountain View, CA, USA; San Francisco, …

4d ago

Solutions Architect, Financial Services Banking USD 152K-241K

Apache Spark | C++ | Cloud Computing | Dask | Deep learning

Senior-level Full Time

US, VA, Remote R

4d ago

Applied Scientist, Neuron ARG, Annapurna ML USD 171K-222K

Accelerator design | Artificial Intelligence | Compiler design | Deep learning | Intermediate representation

401k matching | Adoption and surrogacy reimbursement | Employee assistance program | Flexible spending accounts | Health insurance

Mid-level Full Time

Seattle, Washington, USA

4d ago

Senior Applied Scientist, Annapurna ML USD 192K-260K

Artificial Intelligence | Compiler | Deep learning | Generative AI | Intermediate representation

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

Senior-level Full Time

Seattle, Washington, USA

4d ago

GEN AI Specialist CAD 90K-100K

AWS SageMaker | Azure AI | CUDA | ChromaDB | Clustering

Company-Paid Holidays | Disability insurance | Employee assistance program | Health insurance | Life insurance

Senior-level Full Time

Montreal, CA

4d ago

Machine Learning Research Engineer USD 100K-150K

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

Career growth | Direct W2 employment | Remote work

Senior-level Full Time

United States - Remote R

4d ago

AI Research Engineer USD 100K-150K

Accelerators | Computer Vision | Deep learning | Distributed Training | JAX

Senior-level Full Time

United States - Remote R

4d ago

Executive Director (Senior Lead Securities Python Quantitative Developer ) USD 143K-224K

ALM | Agile | C++ | CPython | CUDA Python

On-site work | Travel up to 10 percent | Visa sponsorship

Senior-level Full Time

144039-NC-550 S Tryon, United States

4d ago

Senior AI Infrastructure Software Engineer - DGX Cloud USD 184K-287K

AI Inference | AI Training | C++ | Cause analysis | Data Infrastructure

Equity | Health benefits | Mentorship

Senior-level Full Time

US, CA, Santa Clara R

4d ago

Machine Learning Senior Software Engineer USD 121K-218K

Adversarial Robustness | CI/CD | Cloud infrastructure | Containerization | Content Safety

401k savings plan | Company holidays | Employee assistance program | Flexible Working Program | Healthcare

Senior-level Full Time

United States

4d ago

AI Research Engineer USD 200K-400K

Agent systems | Constrained decoding | Data Curation | Evaluation Harness | Hugging Face

Mid-level Full Time

New York City

4d ago

Principal Scientist, Machine Learning USD 208K-286K

ADMET | Amazon Bedrock | Anthropic | Autogen | Biomolecule modeling

Senior-level Full Time

Cambridge, MA USA

4d ago

Senior ML Engineer (AI Research, Physical AI) GBP 92K-173K

Computer Vision | Control | Deep learning | Distributed Computing | Distributed Training

Career growth and learning opportunities | Collaborative culture | Flexibility | International environment | Ownership

Senior-level Full Time

Amsterdam, Netherlands; Remote - Europe; United … R

4d ago

AI/ML Specialist Solutions Architect A EUR 78K-108K

DevOps | Distributed Training | Docker | Git | Helm

Career development | Continuous learning | High ownership | International team collaboration | Remote work flexibility

Senior-level Full Time

Netherlands

4d ago

AI/ML Specialist Solutions Architect A EUR 78K-108K

DevOps | Distributed Training | Docker | Git | Helm

Career development | Continuous learning | Flexible work options | High autonomy | International collaboration

Senior-level Full Time

Ireland

4d ago

AI/ML Specialist Solutions Architect A CHF 128K-192K

CI/CD | Cloud infrastructure | DevOps | Distributed Training | Docker

Career development | Continuous learning | Flexible work options | High autonomy | International team collaboration

Senior-level Full Time

Switzerland

4d ago

AI/ML Specialist Solutions Architect A EUR 78K-108K

CI/CD | DevOps | Distributed Training | Docker | Git

Career development | Flexible work options | High ownership and autonomy | International team collaboration | Learning opportunities

Senior-level Full Time

France

4d ago

AI/ML Specialist Solutions Architect A EUR 78K-108K

Cloud Computing | DevOps | Distributed Training | Docker | Git

Autonomy | Career development | Continuous learning | Remote work flexibility

Senior-level Full Time

Germany

4d ago

AI/ML Specialist Solutions Architect A EUR 78K-108K

C plus plus | CI/CD | Distributed Training | Docker | Git

Career development | High ownership autonomy | International team collaboration | Learning opportunities | Remote work options

Senior-level Full Time

Spain

4d ago

Senior Staff Research Scientist, Recommendation Systems, DeepMind USD 262K-364K

Artificial Intelligence | Data Mining | Deep learning | JAX | Language Models

Senior-level Full Time

Mountain View, CA, USA

4d ago

Senior Research Engineer, Foundation Model Training Infrastructure USD 224K-356K

C++ | CUDA | Data Preprocessing | Data Processing | Data loading

Senior-level Full Time

US, CA, Santa Clara

4d ago

Senior Inference Technical Product Marketing Manager - Accelerated Computing USD 148K-235K

Accelerated computing | CUDA | Competitive Analysis | Data center | Data center architecture

Senior-level Full Time

US, CA, Santa Clara

4d ago

Senior Deep Learning Performance Architect USD 184K-287K

ASIC architecture | Architecture simulation | C# | C++ | CUBLAS

Senior-level Full Time

US, CA, Santa Clara

4d ago

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