Direct from source · No middlemen
JAX Jobs in San Francisco
49 open positions · Updated 3 hours ago
49 JAX roles across 22 companies, most in AI/ML; typical advertised salary $230k
Who is hiring (22 companies)
Role types
Work arrangement: 19 hybrid · 12 on-site · 18 not stated
Advertised salaries: p25 $200k · median $230k · p75 $335k (from 23 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 49 positions
Search with filters →Join Discord as a Senior Machine Learning Engineer to build ML solutions that enhance user engagement and retention.
Join Harvey AI as a Member of Technical Staff to build infrastructure for ML models in legal work.
Join Reflection AI as a Member of Technical Staff to design and implement data sourcing and curation techniques for AI models.
Join Atoms as an AI/ML Research Intern to bridge high-level AI research with real-world physical actuation for autonomous transport platforms.
Join Preference Model as a Senior ML Infrastructure Engineer to build scalable infrastructure for post-training research on large language models.
Join Preference Model as a Research Engineer to advance self-directed learning in large language models within a fast-paced startup.
Join Preference Model as a Research Engineer to design and build RL environments for advancing machine learning capabilities.
Join Lyft as a PhD Machine Learning Software Engineer Intern to tackle open research problems impacting millions of riders.
Join Anthropic as a Research Engineer/Scientist to develop next-generation RL algorithms and architectures at frontier scale.
Join Chai Discovery as a Research Engineer to optimize ML infrastructure for drug discovery in a fast-paced, innovative environment.
Join Discord as a Staff Machine Learning Engineer to build ML systems for consumer revenue across various platforms.
Join Together AI as a Research Intern to work on advanced machine learning techniques and contribute to innovative model shaping projects.
Join Together AI as a Research Intern to work on advanced machine learning techniques and contribute to innovative model shaping projects.
Join Together AI as a Research Intern to work on cutting-edge distributed inference and optimization for large foundation models.
Join Together AI as a Research Intern to work on cutting-edge agentic AI systems in a collaborative environment.
Join Together AI as a Research Intern to explore frontier AI systems and contribute to innovative projects in agentic AI.
Join Together AI as a Research Intern to work on cutting-edge distributed inference and optimization for large foundation models.
Own the strategy and execution for infrastructure-native products at the intersection of W&B and CoreWeave.
Join Discord as a Senior Machine Learning Engineer to build and deploy models that enhance user safety on the platform.