joblaze

Direct from source · No middlemen

Jax Jobs

51 open positions · Updated 3 days ago

Average salary: 165.9k–266.7k/yr

Showing 20 of 51 positions

Search with filters →
Perplexity
Internship - Search Machine Learning Engineer
Perplexity
JAX Rust PyTorch TensorFlow Python
Indexed 3 days ago Apply →
Poolside AI
Member of Engineering (Scalability) Remote
Poolside AI
C++ Cython Fault Tolerance Reliability NVIDIA GPU
Published 3 months ago Apply →
Perplexity
Internship - Search Machine Learning Engineer
Perplexity
JAX Rust PyTorch TensorFlow Python
Indexed 3 days ago Apply →
Perplexity
Member of Technical Staff (AI Engineer, Applied ML)
Perplexity
JAX Machine Learning Content Summarization Recommendation Systems PyTorch
Indexed 3 days ago Apply →
Google DeepMind
SWE - Grids - Fixed Term Contract - 6 Months - London, UK
Google DeepMind
C++ Google Cloud Platform JAX TPU Python
Published 2 weeks ago Apply →
Poolside AI
Member of Engineering (Pre-training / CUDA) Remote
Poolside AI
NVLink C++ Cython Large Language Models CUTLASS
Published 1 month ago Apply →
Poolside AI
Member of Engineering (Pre-training and inference software) Remote
Poolside AI
C++ Cython Large Language Models CI/CD PyTorch
Published 8 months ago Apply →
Datadog
AI Research Engineer – Datadog AI Research (DAIR) Remote
Datadog
C++ Ray JAX CUDA DeepSpeed
Indexed 3 days ago Apply →
Datadog
AI Research Engineer - Datadog AI Research (DAIR) Remote
Datadog
C++ Ray JAX CUDA DeepSpeed
Indexed 3 days ago Apply →
Perplexity
UK Internship Program Remote
Perplexity
Torch Triton JAX CUDA AI/ML
Indexed 3 days ago Apply →
Anduril
Senior Machine Learning Engineer, Sentry Tower
Anduril
ONNX C++ JAX AI/ML TensorRT
Indexed 3 days ago Apply →
SpaceX
Sr. Software Engineer, Computer Vision Visa
SpaceX
C++ Jupyter Grafana JAX Prometheus
Published 1 month ago Apply →
Perplexity
Member of Technical Staff (AI Inference Engineer)
Perplexity
NVLink CuTe DSL Triton JAX CUTLASS
Indexed 3 days ago Apply →
Labelbox
Applied Research Intern Remote
Labelbox
Multimodal Models LLMs RLAIF JAX DPO
Indexed 3 days ago Apply →
Google DeepMind
Research Scientist, Gemini Safety
Google DeepMind
JAX AI/ML reinforcement learning
Published 1 month ago Apply →
Instacart
Senior Machine Learning Engineer II, Ads Response Prediction Remote
Instacart
Generative Retrieval pandas Spark JAX Mixed Negative Sampling
Published 2 weeks ago Apply →
Google DeepMind
Research Scientist, Gemini Diffusion Visa
Google DeepMind
Large-scale Distributed Training Text Diffusion JAX Machine Learning Diffusion Models
Published 1 month ago Apply →
Lyft
Data Scientist, Algorithms - Lyft Ads Remote
Lyft
Snowflake Spark JAX Presto scikit-learn
Published 4 months ago Apply →
Anthropic
Research Engineer, Interpretability Remote Visa
Anthropic
JAX CUDA Interpretability Java Go
Indexed 3 days ago Apply →
Lyft
Data Scientist, Algorithms - Lyft Ads Remote
Lyft
Snowflake Spark JAX Presto scikit-learn
Published 4 months ago Apply →
Page 1 of 3

Related searches

jax Remote jax in London jax in New York jax in San Francisco jax in Berlin jax in Amsterdam Python Ruby On Rails React Devops Machine Learning Golang