|
|
26 to 33 of 33 JAX Jobs in the City of London
london (city of london), south east england, united kingdom Signify Technology
looking for 3+ years developing deep learning systems with shipped models or published research. Hands-on with Vision Language Models, LLMs or generative models. Strong Python + PyTorch/ JAX skills (debugging, profiling, shipping). Passion for bridging the gap between simulation and reality. This is one of the most exciting and technically challenging roles I’ve see with the More ❯
City of London, London, United Kingdom MediaTek
Engineering or related quantitative fields. Background in wireless communication systems, e.g., OFDM and MIMO technologies. Track record of related publications or successful projects. Proficiency in Deep Learning frameworks (PyTorch, JAX). Familiarity with advanced AI methods (e.g., transformer models). Excellent scientific communication and teamwork skills. Enthusiasm for innovative research with real-world impact. Join us to shape the future More ❯
london (city of london), south east england, united kingdom MediaTek
Engineering or related quantitative fields. Background in wireless communication systems, e.g., OFDM and MIMO technologies. Track record of related publications or successful projects. Proficiency in Deep Learning frameworks (PyTorch, JAX). Familiarity with advanced AI methods (e.g., transformer models). Excellent scientific communication and teamwork skills. Enthusiasm for innovative research with real-world impact. Join us to shape the future More ❯
City of London, London, United Kingdom Humanoid
VLMs, or image/video generative models — architecture, training, and inference. Experience with deep learning infrastructure: streaming datasets, checkpointing & state management, distributed training strategies. Strong Python + PyTorch/ JAX; you can profile, debug numerics, and write maintainable research code. You document experiments clearly and communicate trade‐offs crisply. Nice-to-Have: Robotics or autonomous driving experience. RL for LLMs More ❯
london (city of london), south east england, united kingdom Humanoid
VLMs, or image/video generative models — architecture, training, and inference. Experience with deep learning infrastructure: streaming datasets, checkpointing & state management, distributed training strategies. Strong Python + PyTorch/ JAX; you can profile, debug numerics, and write maintainable research code. You document experiments clearly and communicate trade‐offs crisply. Nice-to-Have: Robotics or autonomous driving experience. RL for LLMs More ❯
City of London, London, United Kingdom Cubiq Recruitment
ego engineering culture focused on shipping. What We’re Looking For Deep experience in Applied Machine Learning and Agentic AI systems . Proficiency in modern ML stacks (Python, PyTorch, JAX, Ray, etc.) and production deployment. Proven ability to move fast, ship code, and bridge research with real-world impact. Experience in early-stage or startup environments is a plus. A More ❯
london (city of london), south east england, united kingdom Cubiq Recruitment
ego engineering culture focused on shipping. What We’re Looking For Deep experience in Applied Machine Learning and Agentic AI systems . Proficiency in modern ML stacks (Python, PyTorch, JAX, Ray, etc.) and production deployment. Proven ability to move fast, ship code, and bridge research with real-world impact. Experience in early-stage or startup environments is a plus. A More ❯
City of London, London, United Kingdom Harrington Starr
Experience 5+ years’ experience in machine learning with a focus on training and inference systems Strong programming expertise in Python and C++ or CUDA Proficiency with PyTorch, TensorFlow or JAX Hands-on experience with GPU acceleration and distributed training (Horovod, NCCL or similar) Background in real-time, low-latency ML pipelines Familiarity with cloud and orchestration technologies Contributions to open More ❯
|
|
