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.
92 open JAX jobs on freehire.
Named alongside JAX
Jobs asking for JAX
Twilio
16 hours agoMachine Learning Engineer
Open 0d RemoteEuropeSpainWho we are At Twilio, we’re shaping the future of communications, all from the comfort of our homes. We deliver innovative solutions to hundreds of thousands of businesses and empower millions of developers worldwide…
Posted 16 hours ago Sign in to see your matchAgentic AI AI AutoGen AWS Azure +23 skillsCloudflare
3 days agoSenior Machine Learning Engineer
NewBe an early applicant HybridSeniorAbout Us At Cloudflare, we are on a mission to help build a better Internet. Today the company runs one of the world’s largest networks that powers millions of websites and other Internet properties for customers…
Posted 3 days ago Sign in to see your matchAI Cloudflare Deep Learning Distributed Systems DNS +17 skillsCuspAI
3 days agoPlatform Engineer
NewBe an early applicant HybridUKFull-timeUnited KingdomAbout CuspAI CuspAI is the frontier AI company on a mission to solve the breakthrough materials needed to power human progress. While nature took billions of years to perfect molecules, we are harnessing AI to unlock…
Posted 3 days ago Sign in to see your matchAI Automation AWS CI/CD Cloud +21 skillsEmbedding VC
4 days agoMember of Technical Staff — Agent Post-Training
New On-siteNorth AmericaFull-timeUnited StatesIntroducing Moonlake, AI for creating world simulations. About Moonlake Moonlake is building the frontier of interactive world models: systems that generate, simulate, and reason over 3D environments for robotics,…
Posted 4 days ago Sign in to see your matchAgentic AI AI Distributed Training Fine Tuning GPU +5 skillsNormal Computing Corporation
5 days agoCompiler Frontend Engineer
New HybridNorth AmericaFull-timeMiddleUnited StatesNormal Computing | Build with Us Normal is an applied AI company solving the hardest problems in AI and silicon. We build foundational hardware and software for the semiconductor industry, critical AI infrastructure,…
Posted 5 days ago Sign in to see your matchAccessibility AI C++ FPGA JAX +7 skillsDiscord
5 days agoSenior Software Engineer, Machine Learning (Safety)
New North AmericaFull-timeSeniorUnited StatesDiscord has a highly engaged community of millions of daily active users who use the platform for many different reasons, but there’s one thing that nearly everyone does: play video games. Discord plays a uniquely…
Posted 5 days ago Sign in to see your matchData Science JAX Machine Learning Python PyTorch +2 skills