Direct from source · No middlemen

Staff JAX Jobs

20 open positions · Updated 15 hours ago

Average salary (USD/year): 231.6k–353.9k/yr
20 roles Staff 20 3–10 yrs experience Greater Seattle Area +12 more Full-time newest 15 hours ago

20 JAX roles across 15 companies, most in AI/ML; 5% fully remote; typical advertised salary $230k

Work arrangement: 1 fully remote · 5 hybrid · 3 on-site · 11 not stated

Advertised salaries: p25 $199.9k · median $230k · p75 $255.5k (from 12 disclosed annual salaries, USD)

Counts are open roles Joblaze currently tracks on company career pages for this exact skill/location; salaries are advertised minimums, annual, converted to USD.

Showing 20 of 20 positions

Search with filters →
Profound

Join Profound as a Post-Training Engineer to enhance AI models for Fortune 500 marketing teams in a fast-paced environment.

Profound New York, New York, United States $230k–$325k/yr Published 1 day ago
Flexible on stack
Harvey

Join Harvey as a Research Engineer to build infrastructure for ML models in a fast-paced, fully remote environment.

Harvey Remote $230k–$450k/yr Published 1 day ago
Flexible on stack 70% coding
Harvey AI

Join Harvey AI as a Member of Technical Staff to build infrastructure for legal AI models in a fast-growing company.

Harvey AI New York $230k–$340k/yr Published 1 day ago
Flexible on stack
Harvey AI

Join Harvey AI as a Member of Technical Staff to build infrastructure for ML models in legal work.

Harvey AI San Francisco $230k–$340k/yr Published 1 day ago
Flexible on stack
Reflection AI

Join Reflection AI as a Member of Technical Staff to design and implement data sourcing and curation techniques for AI models.

Reflection AI San Francisco, CA Published 1 week ago
Flexible on stack
Discord

Join Discord as a Staff Machine Learning Engineer to build ML systems for consumer revenue across various platforms.

Discord San Francisco Bay Area or Remote (U.S.) $272k–$340k/yr Published 1 month ago
Flexible on stack
Coreweave

Own the strategy and execution for infrastructure-native products at the intersection of W&B and CoreWeave.

Coreweave Livingston, NJ / New York, NY / San Francisco, CA / Sunnyvale, CA / Bellevue, WA $188k–$275k/yr Published 9 months ago
Flexible on stack
Anthropic

Join Anthropic as a Staff Software Engineer to build scalable ML infrastructure for AI safety systems.

Anthropic San Francisco, CA $320k–$485k/yr Published 1 month ago
Flexible on stack
Fireworks AI

Join Fireworks AI as a Member of Technical Staff to advance generative AI and multimodal systems through foundational research.

Fireworks AI Greater Seattle Area Published 1 day ago
Flexible on stack
Perplexity AI

Join Perplexity AI as a Technical Staff member to enhance our AI inference engine with cutting-edge technologies.

Perplexity AI San Francisco Published 6 months ago
Flexible on stack
Perplexity AI

Join Perplexity AI as an AI Inference Engineer to optimize and develop our inference engine for various model architectures.

Perplexity AI London Published 6 months ago
Flexible on stack
Inferact

Join Inferact as a TPU performance engineer to optimize vLLM for Google TPUs, enhancing AI inference performance.

Inferact Singapore S$200k–S$400k/yr Published 4 months ago
Flexible on stack
Inferact

Join Inferact as a TPU performance engineer to optimize vLLM for Google TPUs, enhancing AI inference performance.

Inferact San Francisco $200k–$400k/yr Published 4 months ago
Flexible on stack
Arena

Join Arena as a Machine Learning Scientist to evaluate AI models and contribute to impactful research in a collaborative environment.

Arena SF Bay Area Published 10 months ago
Flexible on stack 70% coding
Atoms

Join Atoms as a Staff Machine Learning Engineer to bridge AI research and real-world physical actuation in autonomous transport.

Atoms San Francisco, CA $273k–$345k/yr Published 4 months ago
Flexible on stack 70% coding
Udio

Join Udio as a full-stack scientist to lead quantitative research efforts at the intersection of music and AI.

Udio New York City (Remote possible for exceptional candidates) $250k–$350k/yr Published 9 months ago
Flexible on stack
Freenome

Join Freenome as a Staff Machine Learning Scientist to drive innovative AI research in cancer detection.

Freenome Remote $199.7k–$283.5k/yr Published 3 months ago
Flexible on stack
Fireworks AI

Join Fireworks AI as a Member of Technical Staff to advance generative AI through foundational research and collaboration with top experts.

Fireworks AI San Mateo Published 2 months ago
Flexible on stack