494 jobs for JAX

AIML - Applied ML Engineer, Responsible AI EUR 78K-103K

Data Processing | Data Processing Pipelines | Diffusion Models | Fine Tuning | JAX

Senior-level Full Time

Barcelona

14h ago

Senior Machine Learning Engineer, Agentic Science/Generative Models, AI for Biology & Translation (AIBT) USD 168K-312K

AWS | AWS SageMaker | Agent Orchestration | AgentOps | Agentic Systems

Senior-level Full Time

South San Francisco, United States

1d ago

Lead Software Engineer in Test, Machine Learning Systems USD 234K-319K

Build systems | CI/CD | Dependency management | JAX | Machine Learning

401k matching | Flexible paid time off | Insurance plans | Team building events

Senior-level Full Time

Los Altos, CA R

1d ago

Software Engineer, Machine Learning CAD 175K-200K

C++ | Coding assistants | Computer Vision | Data Lakes | Data Pipelines

Senior-level Full Time

Markham, Ontario, Canada (Hybrid) R

1d ago

Robotics AI Engineer A SGD 107K-161K

Control Theory | Domain Randomization | Dynamics | Imitation Learning | IsaacGym

High visibility | Open source opportunities | Short feedback loops

Mid-level Full Time

Singapore

1d ago

Robotics Researcher, Manipulation A SGD 95K-245K

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

Competitive compensation | Equity | Open source contributions | Research focused culture

Mid-level Full Time

Singapore R

1d ago

AI Research Engineer USD 180K-287K

Applied cryptography | Blockchain Protocols | Cryptographic mechanisms | Distributed Systems | Federated Learning

Mid-level Full Time

Anywhere R

1d ago

Principal Architect, AI/ML USD 182K-251K

AI Agent Builder | AI Evaluation | Agent Builder | Agentic Systems | Artificial Intelligence

Competitive benefits | EU or UK based eligibility | Remote work

Senior-level Full Time

EMEA (Remote) R

1d ago

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

2d 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

3d 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)

3d 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

3d 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, …

3d 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

3d 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

4d 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

4d ago

ML Software Research Engineer USD 143K-264K

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

Mid-level Full Time

Pittsburgh

4d 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

4d 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

4d 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

4d 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

4d 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

4d 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

4d 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

4d 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

4d 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

4d 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

4d 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

4d 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

4d 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

4d 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

4d 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

4d 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, …

4d 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, …

5d ago

Staff Tech Lead Manager, Machine Learning, Simulator Evaluation USD 238K-302K

3D generative models | Anomaly Detection | C++ | Data Pipelines | Data Processing

Annual bonus | Company benefits program | Equity incentive plan

Senior-level Full Time

Mountain View, California, United States; San … R

5d 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

5d 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

5d 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

5d 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

5d 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

5d ago

AI Research Engineer USD 100K-150K

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

Senior-level Full Time

United States - Remote R

5d 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

5d 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

5d 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

5d ago

AI Research Engineer USD 200K-400K

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

Mid-level Full Time

New York City

5d ago

Principal Scientist, Machine Learning USD 208K-286K

ADMET | Amazon Bedrock | Anthropic | Autogen | Biomolecule modeling

Senior-level Full Time

Cambridge, MA USA

5d 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

5d 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

5d 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

5d 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

5d ago

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