1 – 3 of 3 jobs
Reinforcement learning post-training is driving some of the most significant capability gains in AI today. It is the process that teaches a model to reason through hard problems, follow complex instructions, and act as an autonomous
JAX is the framework of choice for the most ambitious AI research, and the datacenters it runs on are evolving fast. NVIDIA supercomputers are becoming more heterogeneous combining GPUs, CPUs, and LPUs, and JAX needs to
We are seeking highly skilled and motivated software engineers to join us and build AI inference systems that serve large-scale models with extreme efficiency. You’ll architect and implement high-performance inference stacks, optimize GPU kernels and compilers, drive