jax Jobs

9 jobs from companies building with AI

agents 134 alignment 168 deep-learning 80 distributed-systems 188 embeddings 47 evaluation 27 fine-tuning 158 go 49 gpu 57 inference 33 infrastructure 82 llm 519 mlops 33 nlp 71 pre-training 41 pytorch 170 reinforcement-learning 93 research 163 search 174 tensorflow 65

Research Engineer — Claude Models

Anthropic · San Francisco, CA · $250k - $450k
llm rlhf pytorch jax transformers alignment python
full-time senior

Research Scientist, Gemini Safety

DeepMind · Zurich, Switzerland
llm reinforcement-learning fine-tuning jax research search
full-time mid

Research Scientist, AnthroKrishi

DeepMind · Bangalore, India
tensorflow computer-vision deep-learning pytorch jax research
full-time mid

Research Engineer, Gemini Post Training, Model Behavior

DeepMind · Mountain View, California, US
fine-tuning llm jax research search
full-time mid

Research Engineer, Machine Learning - Paris/London/Zurich/Warsaw

Mistral AI · Paris
tensorflow jax nlp deep-learning llm pre-training
full-time mid

Research Engineer, Machine Learning

Mistral AI · Palo Alto
gpu pre-training pytorch jax deep-learning llm
full-time mid

Senior ML Systems Engineer, Frameworks & Tooling

Cohere · London
distributed-systems llm pytorch jax nlp gpu
full-time senior

Research Engineer, Performance RL

Anthropic · San Francisco, CA
llm gpu distributed-systems code-generation pytorch jax
full-time mid

Performance Engineer, GPU

Anthropic · San Francisco, CA | New York City, NY | Seattle, WA
llm pytorch jax distributed-systems gpu alignment
full-time mid

Get AI Jobs in Your Inbox

Weekly digest of the best AI developer opportunities.

Agentic API

Build with our API. Let your agents post jobs and apply automatically.

# Search jobs
curl https://aidevboard.com/api/v1/jobs?tags=llm,pytorch

# Register for API access
curl -X POST https://aidevboard.com/api/v1/register/company \
  -d '{"name":"Acme AI","email":"hire@acme.ai"}'