226 to 241 of 241 PyTorch Jobs

TPUs, high-performance networking (NVLink/InfiniBand), Kubernetes/Slurm, and OS internals Expertise in Python and deep experience with modern deep learning frameworks (PyTorch and/or JAX) Advanced degree (MS or PhD) in Computer Science, Machine Learning, Physics, Mathematics, or a related quantitative field, or equivalent industry experience … skills, results orientation, and excellent collaborative communication Reliable and predictable availability Bonus Points Expertise in: CUDA kernel development, Triton/Pallas/CuTe DSLs, PyTorch/JAX internals, XLA optimization, or hardware acceleration (FPGA/ASIC) Knowledge of reinforcement learning, post-training, or fine-tuning techniques Knowledge of financial markets ...

moving parts: rollouts, replay buffers, reward signals, data filtering, policy updates, evaluation loops, and failure analysis Are familiar with deep learning frameworks such as PyTorch or JAX Are proficient in Python, including concurrency, asynchronous programming, multiprocessing, and performance optimization Can debug distributed GPU workloads across CUDA runtime, container runtime, driver … versions, NCCL or equivalent communication layers, networking, storage, scheduling, and checkpointing Have experience with profiling tools across the stack, for example py‐spy, PyTorch profiler, Nsight, perf, tracing, metrics, logs, or custom instrumentation Have experience with inference stacks such as vLLM, SGLang, TensorRT‐LLM, Dynamo, or custom serving infrastructure ...

Who We AreAt JD.com, also known as JINGDONG, we are building one of the world’s most trusted technology and supply chain ecosystems. What started as a bold idea in China has evolved into a ...

About the Team You’ll be joining the JD.com team, which values collaboration, ownership, and a commitment to delivering high-quality results in a fast-paced, dynamic environment. We are seeking a Machine Learning Team ...

Job Description As a Lead Data Scientist, you will spearhead the development of advanced probabilistic models that power real‐time betting markets. The Quantitative Analysis department harnesses data‐driven insights to enhance our betting products ...

Senior NPU SDK Software Engineer to join our SDK team in London. You will be responsible for the Python programming API of our chip, PyTorch customization, and the compiler for end-to-end applications. You will collaborate with Digital Design, Firmware and Applications teams to translate complex specifications into efficient ...

About Huawei Research and Development UK Limited Founded in 1987, Huawei is a leading global provider of information and communications technology (ICT) infrastructure and smart devices. We have 207,000 employees and operate in over ...

About Niantic Spatial At Niantic Spatial, we’re building the future of physical AI. Powered by a proprietary database of over 30 billion posed images, our groundbreaking mapping technology unlocks a new dimension of interaction ...

TestYantra Software Solutions is seeking a Sr. Gen AI Architect with over 7 years of experience in AI/ML architecture. The role involves designing and deploying generative AI models and requires hands-on experience ...

Overview Conduct original research in computer vision, machine learning, and spatial AI. Design, implement, and evaluate novel algorithms for 3D reconstruction, neural scene representations, 3D Gaussian Splatting, geometric deep learning, Structure-from-Motion (SfM), feed ...

building and deploying production ML systems Strong Python software engineering skills Experience training, fine-tuning or deploying modern machine learning models Strong experience with PyTorch and/or JAX Experience building scalable ML infrastructure and inference pipelines Comfortable owning projects end-to-end in fast-moving environments Previous mentoring … technical leadership experience would be advantageous Tech Stack Python * PyTorch * JAX * GPU Training * LLMs * Modern ML Infrastructure Get in touch for more details - McGregor Boyall is an equal opportunity employer and do not discriminate on any grounds. ...

alongside managing cluster orchestration using slurm and Kubernetes while preparing to expand into specialised GPU providers. And finally you will master the stack from pytorch based learning libraries to complex data networking and GPU utilisation. We're looking for an experienced AI Infrastructure or MLOps Engineer who has already built … operated production AI systems rather than someone at the beginning of their career. You'll have strong experience with the PyTorch ecosystem and a solid understanding of modern transformer architectures, including how to train and operate them at scale. Experience working with computer vision models and image-based datasets ...

with other researchers and engineers to accelerate project velocity.### Skills Required* reinforcement learning (RL)* supervised fine-tuning (SFT)* agent-based systems* JAX* Flax* TensorFlow* PyTorch* systems design### Tags:UKLondonShare Job:Application planning## What to evaluate before applying### Visa and relocation signals* Visa sponsorship is not explicitly marked on this role. … agent-based systems, including modeling or applied infrastructure. 2 years of experience with machine learning frameworks such as JAX, Flax, TensorFlow, or PyTorch, and scaling experiments. Preferred qualifications: PhD in Computer Science, Machine Learning, or a related quantitative field. Publications in reinforcement learning (RL), tool-use, or agentic systems ...

across Bumble products, delivering measurable improvements in user engagement and safety Design, develop, and deploy production‐grade models using modern ML frameworks such as PyTorch, ensuring scalability and reliability in high‐traffic environments Build and deploy production AI Agents using raw and fine‐tuned foundational Large Language Models (LLMs), along … hands‐on experience building and deploying large‐scale systems in production environments. Strong proficiency in Python and at least one major ML framework (e.g., PyTorch, TensorFlow), with experience in areas such as recommendation systems, ranking models, or NLP. Expertise in prompting and fine‐tuning Large Language Models (LLMs) and building ...

mission is to make that hardware serve large AI models at rack scale by building and extending the runtime and serving stack that connects PyTorch and JAX down to the metal. This is a whole-stack systems role. You’ll work where the runtime, the network, and the accelerator meet … collective communication, KV-cache and memory management, and reliability across distributed failure domains at cluster scale. Fluency at the framework boundary, connecting accelerators to PyTorch/JAX and serving stacks without treating either as opaque. Whole-stack debugging: end-to-end and timeline tracing, workload replay, and reasoning from architectural ...

Inventing innovative machine learning systems to solve real-world problems in the defence sector Cambridge; to c£70,000 DoE + Benefits This innovative team of engineers and scientists are using machine learning tightly integrated ...