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
- 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...SeniorPerformance
- ...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...SeniorPerformance
- ...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...SeniorPerformanceVisa sponsorshipWork visa
$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....SeniorPerformance$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...SeniorPerformance- 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
- ...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...SeniorPerformanceFlexible hours
$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...SeniorPerformance$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...SeniorPerformanceWork at office$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$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...SeniorPerformanceFull timeTemporary work$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...SeniorShift work$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$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...SeniorPerformanceWork experience placement$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...SeniorPerformanceWorldwide$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...PerformanceRemote workFlexible hours$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...SeniorPerformance$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...SeniorPerformance$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...SeniorRemote work$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...SeniorPerformance- ...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
- ...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
$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...SeniorPerformanceTemporary work$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...SeniorPerformance- ...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
- ...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...SeniorFlexible hours
- ...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
$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...SeniorPerformance- ...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
$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...SeniorPerformance
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!
- senior ai engineer California, MO
- ai ml engineer California, MO
- ai engineer remote California, MO
- ai engineer California, MO
- ai prompt engineer California, MO
- ai developer California, MO
- machine learning ai engineer California, MO
- platform engineer California, MO
- senior platform engineer California, MO
- platform engineering manager California, MO

