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 Manager, Software Engineering - JAX”. A match may be a passing mention rather than the job itself. Titles only.
44 roles across 47 listings · show every listing · page 1 of 2
…NVIDIA seeks a Senior Engineering Manager to define and drive NVIDIA's JAX strategy, coordinating multiple teams and workstreams to ensure JAX delivers peak…
…working with PyTorch or JAX software - 3+ years of engineering team management experience - 7+ years of working directly within engineering teams experience - 3+ years…
…management, traffic engineering, load balancing, and resiliency mechanisms for large-scale RDMA networks. Optimize end-to-end communication performance across networking, GPU, and software…
…RDMA software engineer with a strong background in high-performance networking, distributed communication systems, and systems programming. You will work closely with senior technical…
…RAG systems, fine-tuning, prompt engineering at scale) - Familiarity with ML frameworks such as PyTorch, TensorFlow, or JAX) - Proficiency with AWS services (SageMaker, Bedrock…
…Experience with asset pipeline toolchains supporting URDF, MJCF, USD, or OpenUSD formats for robot and environment model management. Software Engineering IC4 - The typical base…
…BS or MS or equivalent experience in Computer Science, Electrical Engineering, or related field. 12+ years in systems software engineering with hands-on experience…
…to accelerate deep-learning workloads This role is for a senior software engineer in the Compiler team for AWS Neuron. As part of this…
…or hardware resource management - 3+ years of full software development life cycle, including coding standards, code reviews, source control management, build processes, testing, and…
…Software Engineer/Senior: $160,000.00 - $225,000.00/per year Your actual level and base salary will be determined on a case-by…
…engagement and monetization Partner with engineering to ensure models meet production latency and scalability requirements * Collaborate with product managers to frame business problems as…
…This role is for a senior software engineer in the Machine Learning Inference Applications team. This role is responsible for development and performance optimization…
The MLIL DataPlane team is looking for a Senior Software Development Engineer to own the design and implementation of our inference data plane. We…
…Previous experience working with system software for accelerators such as GPUs, DPUs, or FPGAs. Internals of frameworks like PyTorch and JAX. NVIDIA is widely…
…is looking for an experienced Senior Software Engineer to contribute to the design, development, and commercialization of software/firmware solutions for Cloud Edge and…
…providers to have an innovative AI/ML software infrastructure. Work directly with best-in-class engineering teams to secure design wins, address challenges, bring…
…3D perception - Collaborate with cross-functional teams including robotics engineers, software engineers, and product managers to define and deliver perception capabilities - Drive end-to…
…We are looking for engineers with ML software & systems expertise to help b uild the next generation Waymo onboard ML inference engine for Waymo…
…Our mission is to empower Apple’s software engineering teams with efficient, scalable compute. By driving out operational friction and optimizing the broader machine…
…AWS Neuron is hiring a Principal Technical Product Manager to define and drive product strategy for training software on Trainium. This includes distributed training…
…As the Software Development Engineer for the Neuron Runtime Team, you will be responsible for working alongside a team of engineers to develop and…
…g., hardware engineering, software engineering, program management) - Experience working with PyTorch or JAX software, or experience programming with at least one software programming language…
…Working with tech giants to develop and demonstrate solutions based on NVIDIA’s groundbreaking software and hardware technologies. Partnering with Sales Account Managers and…
…JAX, Megatron) Knowledge of BMC/IPMI/Redfish for out-of-band management and hardware lifecycle Background in MLOps tooling or ML platform engineering Join…
…experimental design, and data generation and management. It will require curiosity to tackle totally new scientific problems, engineering skills to implement the pieces we…