joblaze

Direct from source · No middlemen

Senior Jax Jobs

32 open positions · Updated 3 days ago

Average salary: 243.3k–395.4k/yr

Showing 20 of 32 positions

Search with filters →
Perplexity
Member of Technical Staff (AI Engineer, Applied ML)
Perplexity
JAX Machine Learning Content Summarization Recommendation Systems PyTorch
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 →
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 →
Anthropic
Research Engineer, Interpretability Remote Visa
Anthropic
JAX CUDA Interpretability Java Go
Indexed 3 days ago Apply →
Labelbox
Applied Research Engineer, Agents Remote
Labelbox
diffusion RLAIF LLM agents JAX τ-bench
Indexed 3 days ago Apply →
Google DeepMind
Research Scientist, AnthroKrishi
Google DeepMind
JAX Machine Learning AI/ML PyTorch TensorFlow
Published 8 months ago Apply →
Google DeepMind
Research Scientist, Gemini Safety
Google DeepMind
JAX AI/ML reinforcement learning
Published 1 month ago Apply →
Stability AI
Research Scientist – Controlled 3D Generation Remote
Stability AI
Flow Matching JAX Geometry Processing CUDA Machine Learning
Indexed 4 days ago Apply →
Google DeepMind
Research Scientist, Reinforcement Learning
Google DeepMind
JAX AI/ML reinforcement learning PyTorch Flax
Published 1 month ago Apply →
Anthropic
Research Engineer / Research Scientist, Vision Remote Visa
Anthropic
JAX Large Language Models Machine Learning AI/ML reinforcement learning
Indexed 3 days ago Apply →
Anthropic
Research Engineer/Research Scientist, Audio Remote Visa
Anthropic
JAX Large Language Models Diffusion Models Audio Codecs reinforcement learning
Indexed 3 days ago Apply →
Google DeepMind
Research Scientist, AQUA Visa
Google DeepMind
JAX Machine Learning AI/ML reinforcement learning PyTorch
Published 1 month ago Apply →
Labelbox
Applied Research Engineer Remote
Labelbox
Multimodal Models Direct Preference Optimization (DPO) JAX Large Language Models Reinforcement Learning from Human Feedback (RLHF)
Indexed 3 days ago Apply →
Google DeepMind
Research Engineer, Multimodal Generative AI (Image/Video) Visa
Google DeepMind
image synthesis JAX AI/ML PyTorch TensorFlow
Published 1 month ago Apply →
Anthropic
Research Engineer, Pretraining Scaling - London Visa
Anthropic
JAX large-scale distributed systems PyTorch LLM evaluation infrastructure
Indexed 3 days ago Apply →
Google DeepMind
Research Scientist, Multimodal Alignment, Safety, and Fairness
Google DeepMind
JAX multimodal learning reinforcement learning Python Gemax
Published 1 month ago Apply →
Anthropic
Privacy Research Engineer, Safeguards Remote Visa
Anthropic
LLMs JAX Large Language Models Machine Learning t-closeness
Indexed 3 days ago Apply →
Google DeepMind
Research Engineer, Human Understanding
Google DeepMind
JAX multimodal learning reinforcement learning Python Generative AI
Published 3 weeks ago Apply →
Page 1 of 2

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