joblaze

Direct from source · No middlemen

Staff Jax Jobs

4 open positions · Updated 1 month ago

Average salary: 236.9k–353.1k/yr

Showing 4 of 4 positions

Search with filters →
Perplexity
Member of Technical Staff (AI Inference Engineer)
Perplexity London Published 1 month ago
NVLink CuTe DSL Triton JAX CUTLASS
Apply →
Block
Staff Machine Learning Engineer (Modeling), Support

Lead the development of machine learning initiatives to enhance customer support experiences at Block.

Block Seattle, WA, United States of America $276.8k–$415.2k/yr Published 1 day ago
JAX PyTorch TensorFlow Python
Flexible on stack
Apply →
Perplexity
Member of Technical Staff (AI Inference Engineer)
Perplexity San Francisco Published 1 month ago
NVLink CuTe DSL Nsight Systems Triton JAX
Apply →
Google DeepMind
Staff Research Engineer, Applied AI
Google DeepMind Mountain View, California, US $197k–$291k/yr Published 3 months ago
Google Cloud Platform AWS AI/ML PyTorch TensorFlow
Apply →

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