126 to 130 of 130 JAX Jobs in the UK

runs. You Might Be a Good Fit If You Have Strong engineering experience in large-scale distributed training or HPC systems. Deep familiarity with JAX internals, distributed training libraries, or custom kernels/fused ops. Experience with multi-node cluster orchestration (Slurm, Ray, Kubernetes, or similar). Comfort debugging performance … closely with infra, research, and deployment teams. Nice to Have Experience with training LLMs or other large transformer architectures. Contributions to ML frameworks (PyTorch, JAX, DeepSpeed, Megatron, xFormers, etc.). Familiarity with evaluation and serving frameworks (vLLM, TensorRT-LLM, custom KV caches). Experience with data pipeline optimization, sharded datasets ...

systems stack.Build robust systems that ensure reproducible, debuggable, large-scale runs.Qualifications:Strong engineering experience in large-scale distributed training or HPC systems.Deep familiarity with JAX internals, distributed training libraries, or custom kernels/fused ops.Experience with multi-node cluster orchestration (Slurm, Ray, Kubernetes, or similar).Comfort debugging performance issues across … following would also be good to have for this role:Experience with training LLMs or other large transformer architectures.Contributions to ML frameworks (PyTorch, JAX, DeepSpeed, Megatron, xFormers, etc.).Familiarity with evaluation and serving frameworks (vLLM, TensorRT-LLM, custom KV caches).Experience with data pipeline optimization, sharded datasets, or caching strategies.Background ...

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

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 Bonus … 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 ...

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