joblaze

Direct from source · No middlemen

Jax Jobs (Remote)

28 open positions · Updated 2 days ago

Average salary: 225.4k–355.9k/yr

Showing 20 of 28 positions

Search with filters →

Member of Engineering (Scalability) Remote

C++ Cython Fault Tolerance Reliability NVIDIA GPU

Remote (EMEA/East Coast)

Published 3 months ago Apply →

Member of Engineering (Pre-training / CUDA) Remote

NVLink C++ Cython Large Language Models CUTLASS

Remote (EMEA/East Coast)

Published 1 month ago Apply →

Member of Engineering (Pre-training and inference software) Remote

C++ Cython Large Language Models CI/CD PyTorch

Remote (EMEA/East Coast)

Published 8 months ago Apply →

AI Research Engineer – Datadog AI Research (DAIR) Remote

C++ Ray JAX CUDA DeepSpeed

New York, New York, USA

$140k–$400k/yr

Indexed 3 days ago Apply →

AI Research Engineer - Datadog AI Research (DAIR) Remote

C++ Ray JAX CUDA DeepSpeed

New York City, NY, USA

Indexed 3 days ago Apply →

UK Internship Program Remote

Torch Triton JAX CUDA AI/ML

London, UK

Indexed 3 days ago Apply →

Applied Research Intern Remote

Multimodal Models LLMs RLAIF JAX DPO

San Francisco, USA

$35k–$45k/yr

Indexed 3 days ago Apply →

Senior Machine Learning Engineer II, Ads Response Prediction Remote

Generative Retrieval pandas Spark JAX Mixed Negative Sampling

USA

$201k–$253.5k/yr

Published 2 weeks ago Apply →

Data Scientist, Algorithms - Lyft Ads Remote

Snowflake Spark JAX Presto scikit-learn

New York City, NY, USA

$128k–$160k/yr

Published 4 months ago Apply →

Research Engineer, Interpretability Remote Visa

JAX CUDA Interpretability Java Go

San Francisco

$315k–$560k/yr

Indexed 3 days ago Apply →

Data Scientist, Algorithms - Lyft Ads Remote

Snowflake Spark JAX Presto scikit-learn

San Francisco, CA, USA

$128k–$160k/yr

Published 4 months ago Apply →

Applied Research Engineer, Agents Remote

diffusion RLAIF LLM agents JAX τ-bench

San Francisco, USA

$250k–$300k/yr

Indexed 3 days ago Apply →

Research Scientist – Controlled 3D Generation Remote

Flow Matching JAX Geometry Processing CUDA Machine Learning

Remote

Indexed 3 days ago Apply →

Research Engineer / Research Scientist, Vision Remote Visa

JAX Large Language Models Machine Learning AI/ML reinforcement learning

San Francisco, USA

$350k–$850k/yr

Indexed 3 days ago Apply →

Research Engineer/Research Scientist, Audio Remote Visa

JAX Large Language Models Diffusion Models Audio Codecs reinforcement learning

San Francisco

$350k–$500k/yr

Indexed 3 days ago Apply →

Applied Research Engineer Remote

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

San Francisco, USA

$250k–$300k/yr

Indexed 3 days ago Apply →

Principal Applied AI / ML Engineer Remote

Triton JAX AI/ML PyTorch TensorFlow

San Francisco, CA, USA

$308k–$423.5k/yr

Indexed 3 days ago Apply →

Privacy Research Engineer, Safeguards Remote Visa

LLMs JAX Large Language Models Machine Learning t-closeness

San Francisco, USA

$320k–$485k/yr

Indexed 3 days ago Apply →

Senior Site Reliability Engineer, AI Research Remote

Terraform Google Cloud Platform Datadog CI/CD ArgoCD

Australia

Indexed 3 days ago Apply →

Product Manager II - Model Lab Remote

JAX AI/ML PyTorch TensorFlow

New York, New York, USA

$155k–$190k/yr

Indexed 3 days ago Apply →

Page 1 of 2

Related searches

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