joblaze

← Back to jobs

JAX

46 jobs tagged JAX

Research Scientist: Multilingual, Multicultural and Multimodal LLM

Join Google DeepMind as a Research Scientist to advance multilingual and multimodal LLMs, impacting billions of users.

Google DeepMind Tokyo, Japan Published 2 months ago

JAX Machine Learning PyTorch TensorFlow speech processing

Flexible on stack

Apply →

AI Software Engineer (Vehicle Engineering) Visa

Join SpaceX as an AI Software Engineer to develop AI systems that enhance engineering processes for launch vehicles and spacecraft.

SpaceX Hawthorne, CA $125k–$175k/yr Published 1 week ago

JAX Machine Learning reinforcement learning PyTorch TensorFlow

Flexible on stack

Apply →

Sr. Software Engineer (Vehicle Engineering) Visa

Join SpaceX as a Senior Software Engineer to develop AI systems that enhance engineering processes for launch vehicles and spacecraft.

SpaceX Hawthorne, CA $160k–$225k/yr Published 1 week ago

JAX Machine Learning reinforcement learning PyTorch TensorFlow

70% coding

Apply →

Member of Engineering (Reinforcement Learning) Remote

Poolside AI Remote (EMEA/East Coast) Published 1 month ago

JAX Large Language Models reinforcement learning PyTorch Python

Apply →

Member of Engineering (Reinforcement Learning Infrastructure) Remote

Poolside AI Remote (EMEA/East Coast) Published 1 month ago

JAX Large Language Models reinforcement learning PyTorch Python

Apply →

Research Engineer, Machine Learning (RL Velocity) Remote Visa

Anthropic London, UK £370k–£630k/yr Published 1 month ago

JAX AI/ML PyTorch

Apply →

Research Engineer, Machine Learning (RL Velocity) Remote Visa

Anthropic Remote-Friendly (Travel-Required) | San Francisco, CA | New York City, NY $500k–$850k/yr Published 1 month ago

JAX AI/ML PyTorch

Apply →

Member of Engineering (Post-training) Remote

Poolside AI Remote (EMEA/East Coast) Published 1 month ago

JAX Large Language Models PyTorch Transformers Python

Apply →

Sr. Software Engineer, Computer Vision Visa

SpaceX Hawthorne, CA $160k–$225k/yr Published 1 month ago

C++ Jupyter Grafana JAX Prometheus

Apply →

Research Scientist, AQUA Visa

Google DeepMind Bangalore, India Published 3 months ago

JAX Machine Learning AI/ML reinforcement learning PyTorch

Apply →

Research Scientist, Recommendation Systems Remote Visa

Google DeepMind Mountain View, California, US Published 4 months ago

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

Apply →

Research Scientist, Multimodal Alignment, Safety, and Fairness

Google DeepMind Kirkland, Washington, US; Mountain View, California, US; New York City, New York, US $147k–$211k/yr Published 2 months ago

JAX multimodal learning reinforcement learning Python Gemax

Apply →

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

Google DeepMind Kirkland, Washington, US; Seattle, Washington, US $166k–$244k/yr Published 3 months ago

image synthesis JAX AI/ML PyTorch TensorFlow

Apply →

Research Scientist, Reinforcement Learning

Google DeepMind London, UK Published 2 months ago

JAX AI/ML reinforcement learning PyTorch Flax

Apply →

Research Engineer, Multimodal Reasoning For Information Literacy

Google DeepMind Mountain View, California, US $174k–$252k/yr Published 8 months ago

JAX Machine Learning PyTorch TensorFlow Python

Apply →

Research Scientist, Gemini Diffusion Visa

Google DeepMind London, UK Published 2 months ago

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

Apply →

Research Engineer, Human Understanding

Google DeepMind Los Angeles, California, US; Mountain View, California, US $174k–$252k/yr Published 2 months ago

JAX multimodal learning reinforcement learning Python Generative AI

Apply →

Research Scientist, Gemini Safety

Google DeepMind Zurich, Switzerland Published 2 months ago

JAX AI/ML reinforcement learning

Apply →

Research Scientist, Gemini Safety

Google DeepMind Mountain View, California, US Published 2 months ago

JAX AI/ML reinforcement learning

Apply →

Member of Technical Staff (Software Engineer, Applied AI)

Perplexity San Francisco Published 8 months ago

JAX Machine Learning Content Summarization Recommendation Systems PyTorch

Apply →

Internship - Search Machine Learning Engineer

Perplexity Belgrade Published 4 months ago

JAX Rust PyTorch TensorFlow Python

Apply →

Member of Technical Staff (AI Inference Engineer)

Perplexity London Published 1 month ago

NVLink CuTe DSL Triton JAX CUTLASS

Apply →

UK Internship Program Remote

Perplexity London Published 3 months ago

Torch Triton JAX CUDA AI/ML

Apply →

Member of Technical Staff (AI Inference Engineer)

Perplexity San Francisco Published 1 month ago

NVLink CuTe DSL Nsight Systems Triton JAX

Apply →

Internship - Search Machine Learning Engineer

Perplexity London Published 1 month ago

JAX Rust PyTorch TensorFlow Python

Apply →

AI Applied Scientist Remote

Figma San Francisco, CA • New York, NY • United States $153k–$376k/yr Published 5 months ago

C++ JAX Machine Learning AI/ML Java

Apply →

Applied Research Engineer Remote

Labelbox San Francisco Bay Area $250k–$300k/yr Published 1 year ago

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

Apply →

Applied Research Engineer, Agents Remote

Labelbox San Francisco Bay Area $250k–$300k/yr Published 9 months ago

diffusion RLAIF LLM agents JAX τ-bench

Apply →

Applied Research Intern Remote

Labelbox San Francisco Bay Area $35k–$45k/yr Published 9 months ago

Multimodal Models LLMs RLAIF JAX DPO

Apply →

Principal Applied AI / ML Engineer Remote

Faire San Francisco, CA $308k–$423.5k/yr Published 2 months ago

Triton JAX AI/ML PyTorch TensorFlow

Apply →

AI Research Engineer - Datadog AI Research (DAIR) Remote

Datadog Paris, France Published 9 months ago

C++ Ray JAX CUDA DeepSpeed

Apply →

Product Manager II - Model Lab Remote

Datadog New York, New York, USA $155k–$190k/yr Published 9 months ago

JAX AI/ML PyTorch TensorFlow

Apply →

AI Research Engineer - Datadog AI Research (DAIR) Remote

Datadog New York, New York, USA $140k–$400k/yr Published 9 months ago

C++ Ray JAX CUDA DeepSpeed

Apply →

Research Engineer/Research Scientist, Audio Remote Visa

Anthropic San Francisco, CA $350k–$500k/yr Published 4 months ago

JAX Large Language Models Diffusion Models Audio Codecs reinforcement learning

Apply →

Research Engineer, Science of Scaling Remote Visa

Anthropic London, UK £260k–£630k/yr Published 2 months ago

JAX transformer architectures ETL large-scale ML systems reinforcement learning

Apply →

Privacy Research Engineer, Safeguards Remote Visa

Anthropic San Francisco, CA $320k–$485k/yr Published 7 months ago

LLMs JAX Large Language Models Machine Learning t-closeness

Apply →

Research Engineer, Machine Learning (Reinforcement Learning) Remote Visa

Anthropic San Francisco, CA | New York City, NY $500k–$850k/yr Published 1 year ago

C++ Trio JAX async/concurrent programming Rust

Apply →

Research Engineer, Discovery Remote Visa

Anthropic San Francisco, CA $350k–$850k/yr Published 1 year ago

Beam AWS Spark JAX AI/ML

Apply →

ML Infrastructure Engineer, Safeguards Remote Visa

Anthropic San Francisco, CA $320k–$405k/yr Published 11 months ago

AWS Spark JAX Large Language Models transformer architectures

Apply →

Performance Engineer, GPU Remote Visa

Anthropic San Francisco, CA | New York City, NY | Seattle, WA $280k–$850k/yr Published 8 months ago

FP8 NVLink Triton JAX CUTLASS

Apply →

Research Engineer, Interpretability Remote Visa

Anthropic San Francisco, CA $315k–$560k/yr Published 6 months ago

JAX CUDA Interpretability Java Go

Apply →

Research Engineer, Machine Learning (Reinforcement Learning) Remote Visa

Anthropic London, UK £260k–£630k/yr Published 3 months ago

C++ Trio JAX async/concurrent programming Rust

Apply →

Research Engineer, Pretraining Scaling Visa

Anthropic San Francisco, CA $350k–$850k/yr Published 8 months ago

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

Apply →

Research Engineer, Pretraining Scaling - London Visa

Anthropic London, UK £260k–£630k/yr Published 8 months ago

JAX large-scale distributed systems PyTorch LLM evaluation infrastructure

Apply →

Research Engineer, Performance RL Remote Visa

Anthropic San Francisco, CA $350k–$850k/yr Published 2 months ago

Triton JAX CUDA reinforcement learning PyTorch

Apply →

Research Scientist – Controlled 3D Generation Remote

Stability AI Remote Published 6 months ago

Flow Matching JAX Geometry Processing CUDA Machine Learning

Apply →