Sign up to access all features of our service.
  • Job search
  • Favorites
  • Create a CV
    New
  • Salaries
  • Subscriptions

Senior AI Platform Engineer - JAX Performance & DL Tools

NVIDIA

A leading technology company in California is looking for a Senior Engineer to develop its AI platform and enhance performance optimizations in deep learning frameworks using JAX. The ideal candidate will have a solid background in programming with C/C++ and Python, along with extensive experience in machine learning frameworks. Responsibilities include collaborating with researchers to create improved models and building efficient tools. The position offers a competitive salary range based on experience and performance as well as the chance to join a diverse and innovative team. #J-18808-Ljbffr

Vacancy posted 2 days ago
Similar jobs that could be interesting for youBased on the Senior AI Platform Engineer - JAX Performance & DL Tools in California, MO vacancy
  • NVIDIA is seeking a Senior Software Engineer to automate and optimize performance analysis workflows for AI training and inference. You will design tools that scale across engineers and identify performance bottlenecks in AI workloads. The ideal candidate has a strong background... 
    Senior
    Performance

    NVIDIA

    California, MO
    2 days ago
  •  ...ClickUp is seeking a Senior AI Engineer to develop and manage the core AI platform and apply LLMs for intelligent features across the product. The ideal candidate...  ...AI capabilities, ensuring high reliability and performance in production environments. ClickUp offers a supportive... 
    Senior
    Performance

    ClickUp Project Management

    California, MO
    2 days ago
  •  ...are seeking a skilled and experienced Senior AI Engineer – AI Platform to join our ClickUp Engineering team...  ...to ensure high reliability and performance of AI services in production. Integrate...  ...against role‑related criteria. These tools support, but do not replace, human... 
    Senior
    Performance
    Visa sponsorship
    Work visa

    ClickUp Project Management

    California, MO
    2 days ago
  • $152k - $241.5k

     ...unlimited potential of AI to define the next era...  ...Deep Learning Compiler Engineer. NVIDIA is hiring software...  ...leading inference performance, fast build time, reduced...  ...frameworks, such as PyTorch, JAX. GPU kernel authoring...  ...analysis using tools such as Nsight Compute.... 
    Senior
    Performance

    Dormont Manufacturing Company

    California, MO
    3 days ago
  • $184k - $287.5k

     ...NVIDIA is hiring senior engineers to develop its AI platform and more specifically its performance optimizations in deep learning frameworks using JAX, a tool that can differentiate between Python control code and NumPy code. Our goal is to deliver a polished, fast, modular... 
    Senior
    Performance

    NVIDIA

    California, MO
    2 days ago
  • 6AM City, LLC is seeking an experienced engineer to join the Machine Learning Foundations team. This role focuses on building self-service tooling for model lifecycle management while working across the entire machine learning lifecycle. Candidates should have extensive... 
    Senior

    6AM City

    California, MO
    3 days ago
  •  ...next generation of AI-driven analytics —...  ...capabilities as tools, and complete multi...  ...the reliability and performance of agent workflows...  ...team of AI engineers within Teradata's...  ...role reports to the Senior Manager of AI Engineering...  ..., the Agent Platform team, and Product... 
    Senior
    Performance
    Flexible hours

    Teradata Group

    California, MO
    2 days ago
  • $184k - $287.5k

    Join to apply for the Senior High Performance AI Engineer role at NVIDIA NVIDIA has been transforming computer...  ...models, coding agents and developer tooling. Demonstrated ability to optimize...  ..., including on resource‑constrained platforms. Deep expertise in GPU performance... 
    Senior
    Performance

    NVIDIA

    California, MO
    1 day ago
  • $123.24k - $180k

     ...Role As a Sr./Principal AI Engineer within TSMC's...  ...AI-native products and platforms, from initial concept...  ...developer productivity tools, automated workflow optimization...  ...fields in high-performance environments. 5+ years...  ...engineering teams and senior management. Education... 
    Senior
    Performance
    Work at office

    TSMC - Taiwan Semiconductor Manufacturing Company Limited

    California, MO
    1 day ago
  • $144k - $230k

    NVIDIA Gruppe is seeking an experienced AI/ML Engineer in California, MO, to lead the development of innovative AI and LLM-based systems. This...  ...with significant experience in building and deploying AI tools. A competitive salary ranging from $144,000 to $230,000 annually... 
    Senior

    NVIDIA Gruppe

    California, MO
    3 days ago
  • $175k - $190k

     ...Senior Machine Learning Engineer Hybrid, New York • Hybrid, Boston Data...  ...legacy, and we reward performance and impact players with...  ...bridges between ML platform work and building on...  ...infrastructure and tooling necessary to deploy,...  ...for our AI/ML capabilities Create... 
    Senior
    Performance
    Full time
    Temporary work

    Nimblemind.ai

    California, MO
    2 days ago
  • $224k - $356.5k

     ...NVIDIA’s AI Developer Tools organization is seeking a Senior Research Engineer to join our Research team, where we build the AI coding agents, models, datasets, and evaluations at the heart of NVIDIA’s strategy to put AI-powered coding tools in the hands of every CUDA... 
    Senior
    Shift work

    Dormont Manufacturing Company

    California, MO
    2 days ago
  • $144k - $230k

     ...will be doing: Build and implement robust AI/ML tools capable of analyzing production data to...  ...AI Frameworks, recommending the best platforms, toolsets, and architectural approaches...  ...S. in Computer Science, Statistics, or Engineering (or equivalent experience), and 5+... 
    Senior

    NVIDIA Gruppe

    California, MO
    3 days ago
  • $184k - $287.5k

     ...GPUs are at the core of modern AI infrastructure, from training...  ...much as hardware, and compiler engineering is a big part of what makes it...  ...for generating high-performance GPU kernels from frontend compiler...  ...for GPU kernel generation for DL compiler and framework integration... 
    Senior
    Performance
    Work experience placement

    NVIDIA

    California, MO
    2 days ago
  • $148.7k - $297.3k

     ...groundbreaking health platform that combines...  ...platforms. We are seeking a Senior Manager, Platform Engineering. You are a platform...  ...the reliability, performance, scalability, and cost...  ...automation, and the tooling that enables Lingo’s...  ...tooling, including AI driven anomaly detection... 
    Senior
    Performance
    Worldwide

    Abbott Laboratories

    California, MO
    3 days ago
  • $160k - $180k

     ...professionals with cutting-edge AI and automation. This model...  ...Join the Facet Security & Platform Engineering Team? Reinvent the Experience...  ..., incident response, and performance monitoring. Basic Qualifications...  .../Node.js) to build custom tooling and integrations. Preferred... 
    Performance
    Remote work
    Flexible hours

    Facet

    California, MO
    4 days ago
  • $215.3k - $312.3k

     ..., LLC is looking for a seasoned professional to lead their engineering team focused on AI and machine learning solutions. In this hybrid role based in California, you will design and implement high-performance systems while collaborating with cross-functional teams to... 
    Senior
    Performance

    6AM City

    California, MO
    2 days ago
  • $152k - $241.5k

     ...NVIDIA is seeking a Senior Software Engineer – AI Inference to contribute to open-source LLM serving. You will optimize inference engines like vLLM and SGLang to ensure they perform best on NVIDIA GPUs. This role involves implementing runtime improvements, profiling... 
    Senior
    Performance

    NVIDIA

    California, MO
    2 days ago
  • $130.9k - $198k

    A leading innovative technology firm seeks a Senior Software Engineer to join their AI Platform team. Your role will involve designing scalable AI-driven systems and collaborating with cross-functional teams to implement core platform capabilities. Candidates should have... 
    Senior
    Remote work

    Samsara

    California, MO
    1 day ago
  • $152k - $241.5k

     ...looking for a Senior Deep Learning Software Engineer, TensorRT Performance! NVIDIA is seeking...  ...Generative AI, Recommenders and...  ...that have put DL into every software...  ...(e.g. PyTorch, JAX, TensorFlow,...  ...effective computing platform driving our...  ...NVIDIA uses AI tools in its... 
    Senior
    Performance

    NVIDIA

    California, MO
    4 days ago
  •  ...ServiceTitan, Inc. is seeking an experienced engineer to build and implement core services for the Agent OS platform. This role involves writing production code, debugging, and collaborating across teams to deliver reliable APIs and services. Candidates should possess... 
    Senior

    ServiceTitan

    California, MO
    2 days ago
  •  ...Sunroom Rentals is seeking a Senior Staff Engineer to join our AI and Data Platform team in the United States, specifically in California. This role involves working on impactful AI products that cater to global enterprise customers. Candidates should have robust engineering... 
    Senior

    Sunroom Rentals

    California, MO
    2 days ago
  • $149k - $193.5k

     ...creating a unified software platform for over 10 million new...  ...it. Role Summary As a Senior Staff Embedded Platform Graphics Engineer, you will be at the...  ...functional teams to ensure high performance and seamless integration...  .... Use profiling tools to analyze graphics performance... 
    Senior
    Performance
    Temporary work

    6AM City

    California, MO
    2 days ago
  • $184k - $287.5k

     ...NVIDIA seeks a senior engineer to develop and evolve the NIM Platform SDK for AI inference microservices. The role requires expertise in cloud-native microservice architecture and building high-performance software solutions. Candidates should possess 8+ years of experience... 
    Senior
    Performance

    NVIDIA

    California, MO
    2 days ago
  •  ...Yasp in California is seeking a Software Developer Platform to enhance the YASP AI Platform through development and testing. This role involves...  ...development and a dynamic work environment are available. Join us in making a real impact in AI performance! #J-18808-Ljbffr... 
    Performance

    Yasp

    California, MO
    2 days ago
  •  ...Serko Ltd in California is seeking two Senior Full Stack Engineers to lead the development of next-generation AI products. You will craft intuitive internal UIs, design...  ...experience in Python, React, and cloud platforms, contributing to a collaborative engineering environment... 
    Senior
    Flexible hours

    Serko Ltd

    California, MO
    2 days ago
  •  ...A technology company in California is seeking a Senior Software Engineer to lead the design and development of an innovative AI platform. Responsibilities include mentoring junior engineers and collaborating with cross-functional teams while developing scalable cloud... 
    Senior

    JBA International

    California, MO
    2 days ago
  • $148.7k - $297.3k

     ...Abbott Laboratories is seeking a Senior Manager for Platform Engineering to lead and develop a team responsible for the reliability, performance, and scalability of their cloud environment. The ideal candidate will have over 12 years of experience in cloud infrastructure... 
    Senior
    Performance

    Abbott Laboratories company

    California, MO
    2 days ago
  •  ...A premier journalism company in New York City is seeking a Senior Software Engineer for AI Platforms and Products. The role involves designing and maintaining backend services for AI applications, ensuring system reliability and scalability. Candidates must have over 5... 
    Senior

    New York Times

    California, MO
    2 days ago
  • $184k - $287.5k

    A leading tech company is seeking a Senior Performance Engineer in California to enhance AI system performance and datacenter applications. The role requires extensive experience in accelerated computing, deep learning frameworks, and cloud/container architecture. Applicants... 
    Senior
    Performance

    NVIDIA

    California, MO
    4 days ago

Do you want to receive more vacancies?

Subscribe and receive similar vacancies to Senior AI Platform Engineer - JAX Performance & DL Tools. Be the first to apply!