joblaze

← Back to jobs

JAX

45 jobs tagged JAX

Sr. Software Engineer, Computer Vision Visa

SpaceX

C++ Jupyter Grafana JAX Prometheus

USA

$160k–$225k/yr

Published 1 month ago Apply →

Research Engineer, Human Understanding

Google DeepMind

JAX multimodal learning reinforcement learning Python Generative AI

USA

$174k–$252k/yr

Published 3 weeks ago Apply →

Research Scientist, Gemini Safety

Google DeepMind

JAX AI/ML reinforcement learning

Zurich, Switzerland

Published 4 weeks ago Apply →

Research Engineer, Post-Training for Code Security Analysis

Google DeepMind

JAX AI/ML PyTorch TensorFlow

Mountain View, California, US

$141k–$291k/yr

Published 4 months ago Apply →

Research Scientist, Reinforcement Learning

Google DeepMind

JAX AI/ML reinforcement learning PyTorch Flax

London, UK

Published 1 month ago Apply →

Research Engineer, Information Quality

Google DeepMind

JAX Machine Learning PyTorch TensorFlow Python

Mountain View, California, US

$174k–$252k/yr

Published 6 months ago Apply →

Research Scientist, Multimodal Generative AI, Google DeepMind Visa

Google DeepMind

JAX AI/ML PyTorch TensorFlow Python

Singapore

Published 8 months ago Apply →

Research Scientist, Gemini Diffusion Visa

Google DeepMind

Large-scale Distributed Training Text Diffusion JAX Machine Learning Diffusion Models

London, United Kingdom

Published 1 month ago Apply →

Research Scientist, AnthroKrishi

Google DeepMind

JAX Machine Learning AI/ML PyTorch TensorFlow

Bangalore, India

Published 8 months ago Apply →

SWE - Grids - Fixed Term Contract - 6 Months - London, UK

Google DeepMind

C++ Google Cloud Platform JAX TPU Python

London, UK

Published 2 weeks ago Apply →

Research Engineer, Multimodal Generative AI (Image/Video) Visa

Google DeepMind

image synthesis JAX AI/ML PyTorch TensorFlow

Seattle, WA, USA

$166k–$244k/yr

Published 1 month ago Apply →

Research Scientist, Multimodal Alignment, Safety, and Fairness

Google DeepMind

JAX multimodal learning reinforcement learning Python Gemax

Kirkland, Washington, US; Mountain View, California, US; New York City, New York, US

$147k–$211k/yr

Published 1 month ago Apply →

Research Scientist, AI-powered Scientific Discovery

Google DeepMind

JAX Large Language Models AI/ML reinforcement learning PyTorch

Montreal, Canada

Published 1 month ago Apply →

Research Scientist, AQUA Visa

Google DeepMind

JAX Machine Learning AI/ML reinforcement learning PyTorch

Bangalore, India

Published 1 month ago Apply →

Research Scientist, Recommendation Systems Remote Visa

Google DeepMind

JAX Large Language Models AI/ML Supervised Fine-Tuning reinforcement learning

Mountain View, USA

Published 3 months ago Apply →

Research Scientist, Gemini Safety

Google DeepMind

JAX AI/ML reinforcement learning

Mountain View, California, US

Published 1 month ago Apply →

Internship - Search Machine Learning Engineer

Perplexity

JAX Rust PyTorch TensorFlow Python

London, UK

Indexed 1 day ago Apply →

Member of Technical Staff (AI Inference Engineer)

Perplexity

NVLink CuTe DSL Nsight Systems Triton JAX

San Francisco

Indexed 1 day ago Apply →

Member of Technical Staff (AI Engineer, Applied ML)

Perplexity

JAX Machine Learning Content Summarization Recommendation Systems PyTorch

San Francisco

Indexed 1 day ago Apply →

Member of Technical Staff (AI Inference Engineer)

Perplexity

NVLink CuTe DSL Triton JAX CUTLASS

London

Indexed 1 day ago Apply →

UK Internship Program Remote

Perplexity

Torch Triton JAX CUDA AI/ML

London, UK

Indexed 1 day ago Apply →

Internship - Search Machine Learning Engineer

Perplexity

JAX Rust PyTorch TensorFlow Python

Belgrade, Serbia

Indexed 1 day ago Apply →

AI Applied Scientist Remote

Figma

C++ JAX Machine Learning AI/ML Java

United States

$153k–$376k/yr

Indexed 1 day ago Apply →

Applied Research Intern Remote

Labelbox

Multimodal Models LLMs RLAIF JAX DPO

San Francisco, USA

$35k–$45k/yr

Indexed 1 day ago Apply →

Applied Research Engineer, Agents Remote

Labelbox

diffusion RLAIF LLM agents JAX τ-bench

San Francisco, USA

$250k–$300k/yr

Indexed 1 day ago Apply →

Applied Research Engineer Remote

Labelbox

Multimodal Models Direct Preference Optimization (DPO) JAX Large Language Models Reinforcement Learning from Human Feedback (RLHF)

San Francisco, USA

$250k–$300k/yr

Indexed 1 day ago Apply →

Principal Applied AI / ML Engineer Remote

Faire

Triton JAX AI/ML PyTorch TensorFlow

San Francisco, CA, USA

$308k–$423.5k/yr

Indexed 1 day ago Apply →

Product Manager II - Model Lab Remote

Datadog

JAX AI/ML PyTorch TensorFlow

New York, New York, USA

$155k–$190k/yr

Indexed 1 day ago Apply →

AI Research Engineer - Datadog AI Research (DAIR) Remote

Datadog

C++ Ray JAX CUDA DeepSpeed

New York City, NY, USA

Indexed 1 day ago Apply →

AI Research Engineer – Datadog AI Research (DAIR) Remote

Datadog

C++ Ray JAX CUDA DeepSpeed

New York, New York, USA

$140k–$400k/yr

Indexed 1 day ago Apply →

Research Engineer, Pretraining Scaling Visa

Anthropic

ML frameworks JAX model operations AI/ML large-scale distributed systems

San Francisco, USA

$350k–$850k/yr

Indexed 1 day ago Apply →

Research Engineer, Pretraining Scaling - London Visa

Anthropic

JAX large-scale distributed systems PyTorch LLM evaluation infrastructure

London, UK

£260k–£630k/yr

Indexed 1 day ago Apply →

Research Engineer, Science of Scaling Remote Visa

Anthropic

JAX transformer architectures ETL large-scale ML systems reinforcement learning

San Francisco

£260k–£630k/yr

Indexed 1 day ago Apply →

Research Engineer, Performance RL Remote Visa

Anthropic

Triton JAX CUDA reinforcement learning PyTorch

San Francisco

$350k–$850k/yr

Indexed 1 day ago Apply →

Research Engineer/Research Scientist, Audio Remote Visa

Anthropic

JAX Large Language Models Diffusion Models Audio Codecs reinforcement learning

San Francisco

$350k–$500k/yr

Indexed 1 day ago Apply →

Research Engineer, Interpretability Remote Visa

Anthropic

JAX CUDA Interpretability Java Go

San Francisco

$315k–$560k/yr

Indexed 1 day ago Apply →

Privacy Research Engineer, Safeguards Remote Visa

Anthropic

LLMs JAX Large Language Models Machine Learning t-closeness

San Francisco, USA

$320k–$485k/yr

Indexed 1 day ago Apply →

Research Engineer, Machine Learning (Reinforcement Learning) Remote Visa

Anthropic

C++ Trio JAX async/concurrent programming Rust

San Francisco

$500k–$850k/yr

Indexed 1 day ago Apply →

Research Engineer, Discovery Remote Visa

Anthropic

Beam AWS Spark JAX AI/ML

San Francisco, USA

$350k–$850k/yr

Indexed 1 day ago Apply →

ML Infrastructure Engineer, Safeguards Remote Visa

Anthropic

AWS Spark JAX Large Language Models transformer architectures

San Francisco, USA

$320k–$405k/yr

Indexed 1 day ago Apply →

Research Engineer / Research Scientist, Vision Remote Visa

Anthropic

JAX Large Language Models Machine Learning AI/ML reinforcement learning

San Francisco, USA

$350k–$850k/yr

Indexed 1 day ago Apply →

Performance Engineer, GPU Remote Visa

Anthropic

FP8 NVLink Triton JAX CUTLASS

San Francisco

$280k–$850k/yr

Indexed 1 day ago Apply →

Research Engineer, Machine Learning (Reinforcement Learning) Remote Visa

Anthropic

C++ Trio JAX async/concurrent programming Rust

San Francisco

£260k–£630k/yr

Indexed 1 day ago Apply →

Senior Machine Learning Engineer, Sentry Tower

Anduril

ONNX C++ JAX AI/ML TensorRT

USA

$220k–$330k/yr

Indexed 1 day ago Apply →

Research Scientist – Controlled 3D Generation Remote

Stability AI

Flow Matching JAX Geometry Processing CUDA Machine Learning

Remote

Indexed 1 day ago Apply →