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.
102 roles across 108 listings · show every listing · page 1 of 5
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…
…Engage directly with research, software engineering, hardware engineering, and product teams to deliver end-to-end solutions. You have: 7+ years of experience in…
…We are looking for researchers and software engineers passionate about developing ML techniques for evaluation systems and driving performance improvements across our technology stack…
…from data ingestion and feature engineering through to model development, validation, and deployment with guidance from senior engineers on complex architectural decisions Instrument AI…
…engineers to optimize, validate, and deploy large language models. Develop and maintain a world-class model compiler framework, FTL . Write robust, high quality software…
…working with PyTorch or JAX software - 3+ years of engineering team management experience - 7+ years of working directly within engineering teams experience - 3+ years…
We're looking for outstanding AI systems software engineers to develop groundbreaking technologies across the inference systems software stack. Our team builds core AI…
…How do you mathematically prove that a virtual world is "real" ? We are seeking visionary machine learning engineers and researchers to architect the scalable…
…Designing user facing features in JAX and related libraries and other general software engineering work. Working closely with GPU hardware engineering teams to design…
…or engineering applications targeting GPUs Strong fundamentals in programming and software design, especially in Python and C++ Familiarity with computational physics or engineering simulation…
Oracle is seeking a highly experienced Lead Principal Software Engineer (IC5) to help define and build the next generation of AI networking infrastructure powering…
…RDMA software engineer with a strong background in high-performance networking, distributed communication systems, and systems programming. You will work closely with senior technical…
…Proficiency in Python and modern C++, with strong software engineering fundamentals (version control, testing, CI/CD). Deep understanding of 3D geometry, camera models, and…
…JAX, TensorRT) Familiarity with advanced optimizations and SW/HW co-design in LLM training and inference Exposure to using AI to accelerate SW engineering…
…RAG systems, fine-tuning, prompt engineering at scale) - Familiarity with ML frameworks such as PyTorch, TensorFlow, or JAX) - Proficiency with AWS services (SageMaker, Bedrock…
…Software Engineering IC4 - The typical base pay range for this role across the U.S. is USD $119,800 - $234,700 per year. There…
…BS or MS or equivalent experience in Computer Science, Electrical Engineering, or related field. 12+ years in systems software engineering with hands-on experience…
…Engineering, or related field (or equivalent experience). 3+ years of relevant industry experience. Strong programming skills in Python and C++. Experience in software performance…
…to accelerate deep-learning workloads This role is for a senior software engineer in the Compiler team for AWS Neuron. As part of this…
…alongside a talented engineering team, contributing to leading-edge design and research in compiler technology and deep-learning systems software. Additionally, you'll collaborate…
…Software Engineer/Senior: $160,000.00 - $225,000.00/per year Your actual level and base salary will be determined on a case-by…
…systems such as Hadoop, Spark etc. - Experience working with PyTorch or JAX software, or experience with Machine Learning and Large Language Model fundamentals, including…
…This role is for a senior software engineer in the Machine Learning Inference Applications team. This role is responsible for development and performance optimization…
…We are looking for a Senior Software Engineer to join the cuEquivariance team — an NVIDIA library that accelerates geometric neural networks on NVIDIA GPUs…