76 to 83 of 83 Permanent Google JAX Jobs

Join us to push the boundaries of AI and contribute to groundbreaking discoveries in a collaborative, fast-paced environment. You’ll have the opportunity to advance your career by working with the Global Technology Applied ...

design and scale infrastructure that powers the future of AI-driven engineering? Apply now. Responsibilities Design and scale training environments using PyTorch Distributed, JAX, or NVIDIA NeMo across multi-node/multi-GPU clusters. Manage a mixed-compute strategy spanning public clouds (AWS/Azure) and sovereign industrial clouds … that validates physical consistency laws. Skills/Must have Orchestration: Expert-level Kubernetes (AKS/EKS) is essential. ML Frameworks: Strong proficiency in PyTorch, JAX, or NVIDIA NeMo. HPC/Data: Solid experience with Python, Go, and distributed data tools (Dask/Spark). Background: Experience in AI research ...

CoreWeave 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 AI ...

years of experience working in frontier AI research labs on pre-training or post-training teams. Experience writing TPU/GPU kernels (e.g., JAX, CUDA) to optimize model performance and real-time inference. Experience designing, training, and scaling generative pixel or real-time video architectures. Track record of cross … expect applicants to be experts in all fields simultaneously. Artificial intelligence will be one of humanity’s most transformative inventions. At Google DeepMind, we are a pioneering AI lab with exceptional interdisciplinary teams focused on advancing AI development to solve complex global challenges and accelerate high-quality product innovation ...

hackajob is partnering directly with Google to hire for this role. We are looking for a Research Scientist to develop multimodal generative models and advance the frontier of world modeling research. This role focuses on applying diffusion models to real-time video generation, joining our core modeling team … experience. We do not expect applicants to be experts in all fields simultaneously. Artificial intelligence will be one of humanitys most transformative inventions. At Google DeepMind, we are a pioneering AI lab with exceptional interdisciplinary teams focused on advancing AI development to solve complex global challenges and accelerate high ...

Genesis-World works and is used Opinions about API design Bonus: experience with Drake, MuJoCo, Newton, Isaac, RaiSim, Jiminy, or Brax Bonus: PyTorch, Triton, JAX, or Numba Bonus: 3D tooling and interactive editors such as Maya, Blender, or Houdini Bonus: serialization formats such as USD, glTF, MJCF, or URDF … Soft Skills Community Communication Issue Triaging Industry Keywords Simulation Physics Solvers Robotics Interactive Editors 3D Tooling Tools & Technologies CUDA AMD Apple Metal PyTorch Triton JAX Numba Maya Blender Houdini #J-18808-Ljbffr ...

Here at Humanoid, we believe in a future where robots amplify human potential. That’s why we’ve set out on a mission to build the world’s most capable, commercially-scalable, and safe humanoid ...

Member of Technical Staff, Training Infrastructure — Inherent (London) At Inherent, we are on a mission to build AI that recursively self‐improves to discover new knowledge. Scientific advances are the backbone of our economic, technological ...