Jobs
Indexed directly from employers. Every age is their own publish date.
Indexed directly from employers. Every age is their own publish date.
Indexed directly from employers. Every age is their own publish date.
Searching titles and descriptions for “Senior Software Engineer, JAX”. A match may be a passing mention rather than the job itself. Titles only.
157 roles across 172 listings · show every listing · page 1 of 7
NVIDIA is hiring senior engineers to develop its AI platform and more specifically its performance optimizations in deep learning frameworks using JAX, a tool…
…NVIDIA seeks a Senior Engineering Manager to define and drive NVIDIA's JAX strategy, coordinating multiple teams and workstreams to ensure JAX delivers peak…
Google's software engineers develop the next-generation technologies that change how billions of users connect, explore, and interact with information and one another…
…Working across the stack from PyTorch till the hardware-software boundary, our engineers build systematic infrastructure, innovate new methods and create high-performance kernels…
…This senior software engineering role is part of the Machine Learning Inference Applications team and focuses on delivering high-performance model inference solutions for…
…Engineering Group, Engineering Group > Software Engineering General Summary: Qualcomm is seeking an exceptional Senior Director, Software Engineering to lead software architecture, development, and execution…
…Collaborate with other scientists, engineers, and teams! You are an ideal candidate if you have: - Extremely strong software engineering skills. - Value test-driven development…
…Engineering Group, Engineering Group > Software Test Engineering General Summary: About Us Qualcomm is growing its presence in Riyadh and is hiring Data Centre Engineers…
…JOB SUMMARY The Senior Reinforcement Learning Engineer is a key, hands-on role focused on achieving state-of-the-art performance on our humanoid…
…In this hybrid role, you will report to a Senior Research Scientist. You will: Develop and extend cutting-edge research in robotics and machine…
…Working across the stack from PyTorch till the hardware-software boundary, our engineers build systematic infrastructure, innovate new methods and create high-performance kernels…
…In this hybrid role, you will report to a Senior Research Scientist. You will: Build and operate the petabyte-scale data systems and ML…
…PyTorch, JAX, Ray), and/or inference and deployment environments (e.g. TRTLLM, vLLM, SGLang). Proficient in Python programming, software design, debugging, performance analysis, test…
…software engineers, and machine learning engineers passionate about delivering innovative technologies that impact millions of users. We are looking for a senior engineer excited…
The MLIL DataPlane team is looking for a Senior Software Development Engineer to own the design and implementation of our inference data plane. We…
…We are looking for a Senior Software Development Engineer to lead the architecture and delivery of core ML platform capabilities. You will solve problems…
We are looking for an experienced and highly motivated software professional to work on pioneering initiatives and projects at the intersection of CUDA and…
…In this hybrid role, you will report to a Senior Director of AI Foundations You will: Own the data recipe for Waymo’s Foundation…
…NVIDIA is hiring a Senior AI Compiler Engineer. GPUs are driving rapid progress in deep learning—from LLMs and generative AI to recommendation, vision…
…In this hybrid role, you will report to a Senior Staff Software Engineer. You will: Partner with foundation model training teams to determine optimal…
…learning fundamentals, applied machine learning experience, and strong software engineering skills. As Senior Machine Learning Engineer for the Search and Knowledge Quality team, you…
…As a Staff/Senior Machine Learning Engineer, you play a critical role in developing world-class Search and Q&A experiences for Apple customers…
…Strong software development skills with proficiency in Python; hands-on experience working with deep learning toolkits like PyTorch, TensorFlow, or JAX. Hands-on practice…
…As such, we are looking for candidates with applied machine learning experience and strong software engineering skills. Build world class large language model at…
…end behavior across GPUs, networking, storage, and software stacks. We are seeking a Senior Performance Engineer to characterize workloads, establish performance baselines, diagnose bottlenecks…