What is JAX?
JAX is a machine learning library used for numerical computing and transforming mathematical functions. It enables automatic differentiation and compiles code to run efficiently on hardware accelerators like GPUs and TPUs.
1,068 open JAX jobs on RoleWatch.
Named alongside JAX
Jobs asking for JAX
- CCoreWeave6 hours ago
Account Solution Architect
NewBe an early applicant UKFull-timeC-levelUnited KingdomCoreWeave is The Essential Cloud for AI™. Built for pioneers by pioneers, CoreWeave delivers a platform of technology, tools, and teams that enables innovators to build and scale AI with confidence. Trusted by leading…
Posted 6 hours ago Sign in to see your matchAccount Executive AI Cloud Confluence CUDA +19 skills - SSpaceXAI8 hours ago
Software Engineer, RL Training Framework
NewBe an early applicant North AmericaUnited StatesSpaceX was founded under the belief that a future where humanity is out exploring the stars is fundamentally more exciting than one where we are not. Today, SpaceX is actively developing the technologies to make this…
Posted 8 hours ago Sign in to see your matchC++ Distributed Systems JAX LLM Observability +3 skills - SSpaceXAI8 hours ago
Software Engineer, RL Inference
NewBe an early applicant North AmericaUnited StatesSpaceX was founded under the belief that a future where humanity is out exploring the stars is fundamentally more exciting than one where we are not. Today, SpaceX is actively developing the technologies to make this…
Posted 8 hours ago Sign in to see your matchC++ CUDA Distributed Systems JAX LLM +5 skills - SSpaceXAI8 hours ago
Machine Learning Engineer, Recommendation Systems
NewBe an early applicant North AmericaUnited StatesSpaceX was founded under the belief that a future where humanity is out exploring the stars is fundamentally more exciting than one where we are not. Today, SpaceX is actively developing the technologies to make this…
Posted 8 hours ago Sign in to see your matchAI ClickHouse CUDA Data Pipelines Deep Learning +6 skills - 442Dot11 hours ago
Sr. Machine Learning Engineer, Autonomy
NewBe an early applicant North AmericaSeniorUnited StatesWe are looking for the best About Us 42dot is a mobility AI company committed to solving mobility challenges with software and AI. As the Global Software Center of Hyundai Motor Group, 42dot pioneers the future of…
Posted 11 hours ago Sign in to see your matchAI Distributed Training JAX Machine Learning Python +1 skills - 442Dot11 hours ago
Machine Learning Engineer, Driving Foundation Models
NewBe an early applicant North AmericaUnited StatesWe are looking for the best About Us 42dot is a mobility AI company committed to solving mobility challenges with software and AI. As the Global Software Center of Hyundai Motor Group, 42dot pioneers the future of…
Posted 11 hours ago Sign in to see your matchAI AWS Azure Cloud C++ +19 skills