joblaze

Direct from source · No middlemen

Pytorch Jax Internals Jobs

33 open positions · Updated 2 months ago

Average salary: 193.1k–371.6k/yr

Showing 20 of 33 positions

Search with filters →

Member of Technical Staff (AI Inference Engineer)

Perplexity London Published 2 months ago

NVLink CuTe DSL Triton JAX CUTLASS

Apply →

Member of Technical Staff (AI Inference Engineer)

Perplexity San Francisco Published 2 months ago

NVLink CuTe DSL Nsight Systems Triton JAX

Apply →

Research Intern, Inference (Fall 2026)

Join Together AI as a Research Intern to work on cutting-edge distributed inference and optimization for large foundation models.

Together AI San Francisco $58–$63/yr Published 5 days ago

JAX CUDA Machine Learning PyTorch Python

Flexible on stack

Apply →

Internship - Search Machine Learning Engineer

Perplexity London Published 2 months ago

JAX Rust PyTorch TensorFlow Python

Apply →

Frontier Agents Intern (Fall 2026)

Join Together AI as a research intern to work on cutting-edge projects in frontier AI systems and agentic behavior.

Together AI San Francisco $58–$63/yr Published 5 days ago

JAX Machine Learning PyTorch Python Transformer

Flexible on stack

Apply →

Internship - Search Machine Learning Engineer

Perplexity Belgrade Published 5 months ago

JAX Rust PyTorch TensorFlow Python

Apply →

Research Intern, Model Shaping (Fall 2026) Remote

Join Together AI as a Research Intern to work on advanced machine learning techniques and contribute to impactful research.

Together AI San Francisco $58–$63/yr Published 2 days ago

JAX Machine Learning PyTorch Transformer Deep Learning

Flexible on stack

Apply →

Research Engineer, Interpretability Remote Visa

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

JAX CUDA Interpretability Java Go

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, 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/Research Scientist, Audio Remote Visa

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

JAX Large Language Models Diffusion Models Audio Codecs reinforcement learning

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 10 months ago

C++ Ray JAX CUDA DeepSpeed

Apply →

Research Scientist, AQUA Visa

Google DeepMind Bangalore, India Published 3 months ago

JAX Machine Learning AI/ML reinforcement learning PyTorch

Apply →

Applied Research Intern, Proactive Intelligence & Customer World Models (PhD / Graduate Co-op) Remote

Join Block as an Applied Research Intern to develop proactive intelligence systems that enhance customer understanding and decision-making.

Block Bay Area, CA, United States of America Published 1 week ago

JAX Machine Learning Causal Reasoning representation learning reinforcement learning

Flexible on stack

Apply →

Applied Research Intern, Proactive Intelligence & Customer World Models (PhD / Graduate Co-op) Remote

Join Block as an Applied Research Intern to develop proactive intelligence systems that enhance customer understanding and decision-making.

Block Toronto, Ontario , Canada Published 1 week ago

JAX Machine Learning Causal Reasoning representation learning reinforcement learning

Flexible on stack

Apply →

Applied Research Intern Remote

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

Multimodal Models LLMs RLAIF JAX DPO

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, 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, Reinforcement Learning

Google DeepMind London, UK Published 3 months ago

JAX AI/ML reinforcement learning PyTorch Flax

Apply →

Page 1 of 2