joblaze

Direct from source · No middlemen

Jax In Jobs

48 open positions · Updated 1 month ago

Average salary: 174.3k–267.1k/yr

Showing 20 of 48 positions

Search with filters →

Internship - Search Machine Learning Engineer

Perplexity London Published 1 month ago

JAX Rust PyTorch TensorFlow Python

Apply →

Member of Engineering (Scalability) Remote

Poolside AI Remote (EMEA/East Coast) Published 4 months ago

C++ Cython Fault Tolerance Reliability NVIDIA GPU

Apply →

UK Internship Program Remote

Perplexity London Published 3 months ago

Torch Triton JAX CUDA AI/ML

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 →

Member of Technical Staff (AI Inference Engineer)

Perplexity London Published 1 month ago

NVLink CuTe DSL Triton JAX CUTLASS

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 →

Internship - Search Machine Learning Engineer

Perplexity Belgrade Published 4 months ago

JAX Rust PyTorch TensorFlow 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 →

Member of Technical Staff (AI Inference Engineer)

Perplexity San Francisco Published 1 month ago

NVLink CuTe DSL Nsight Systems Triton JAX

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 →

Applied Research Intern Remote

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

Multimodal Models LLMs RLAIF JAX DPO

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 →

Research Scientist, Gemini Safety

Google DeepMind Zurich, Switzerland Published 2 months ago

JAX AI/ML reinforcement learning

Apply →

Sr. Software Engineer, Computer Vision Visa

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

C++ Jupyter Grafana JAX Prometheus

Apply →

AI Research Engineer - Datadog AI Research (DAIR) Remote

Datadog Paris, France Published 9 months ago

C++ Ray JAX CUDA DeepSpeed

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 Scientist, Gemini Diffusion Visa

Google DeepMind London, UK Published 2 months ago

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

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 →

Research Scientist, Gemini Safety

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

JAX AI/ML reinforcement learning

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 →

Page 1 of 3