JAX Jobs

47 jobs from companies building with AI · Avg salary $281k (19 with data)

JAX engineering positions. JAX is Google's framework for high-performance ML research, favored for its functional design, XLA compilation, and multi-device parallelism capabilities.

AI/ML Engineer

Brain Co. · San Francisco, CA
pytorch tensorflow data-pipeline llm jax healthcare generative-ai machine-learning
Remote full-time mid 9 months ago

AI/ML Engineer, Deployed

Brain Co. · Abu Dhabi
pytorch generative-ai healthcare llm tensorflow jax data-pipeline payments
Hybrid full-time mid 10 months ago

Quantitative Researcher - Machine Learning

Point72 · New York, NY
jax tensorflow llm reinforcement-learning research machine-learning
On-site full-time mid 11 months ago

Applied AI Research Engineer

Netic · San Francisco, CA
llm jax search pytorch agents data-pipeline research
On-site full-time mid 11 months ago

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

Mistral AI · Paris, France
pytorch llm search deep-learning nlp tensorflow jax gpu
On-site full-time mid 1 year ago

ML Infra Engineer

Physical Intelligence · San Francisco, CA
pytorch jax distributed-systems gpu generative-ai cloud robotics infrastructure
On-site full-time mid 1 year ago

Member of Technical Staff, Staff Physicist, Quantum Information and AI

First Principles · Remote
jax reinforcement-learning agents cloud fine-tuning pytorch
Remote full-time lead 5 months ago

Weekly AI Jobs Digest

Top new roles from 27+ companies. Curated, not scraped. One email, every Monday.

No spam. Unsubscribe anytime.

Hiring AI engineers?

Post the role first. Your company profile and analytics connect from the employer flow.

Post a Job See Pricing

Agentic API & MCP Server

Wire AI Dev Jobs into your agent at build time — MCP server live, REST API public for discovery, free API keys for recurring search.

# Add as MCP server
claude mcp add --transport http aidevjobs https://aidevboard.com/mcp

# Or hit the REST API
curl https://aidevboard.com/api/v1/jobs?tags=llm,pytorch

13 tools via com.aidevboard/jobs · Free keyed access · Pro $49/mo →