Senior AI Platform Engineer - JAX & Performance (Equity)
NVIDIA Gruppe
NVIDIA is seeking senior engineers to advance its AI platform, focusing on performance optimizations in deep learning frameworks using JAX. The role emphasizes modular, fast, and coordinated tooling to handle data, training and analysis for diverse DL solutions.
Strong programming, system design, and communication skills are essential as you collaborate with researchers and engineers. You will work across CPU/GPU architectures, build scalable distributed components, and contribute to
#J-18808-LjbffrVacancy posted 9 hours ago
Similar jobs that could be interesting for youBased on the Senior AI Platform Engineer - JAX & Performance (Equity) in Santa Clara, CA vacancy
- ...NVIDIA is hiring senior engineers to develop its AI platform and optimize deep learning frameworks using JAX. You will join a fast, modular, and scalable effort focused on data... ...design core JAX components, drive peak performance on NVIDIA hardware, collaborate with AI researchers...SeniorPerformance
NVIDIA AI
Santa Clara, CA9 hours ago $184k - $287.5k
...the unlimited potential of AI to define the next era of computing... ...Group (SCG) is seeking Senior AI Platform Engineers. They will set the... ...for security, reliability, performance, and evolution.Leading the... ...will also be eligible for equity and benefits.Applications for...Employment EquitySeniorPerformanceFull time$151.8k - $265.35k
...content effortlessly. The AI for Engineering team builds a... ...production-grade AI platform that powers creativity... ...memory.Develop high-performance inference and runtime... ...adaptive AI systems.Mentor senior engineers in modern... ...the form of a new hire equity award.State-Specific...Employment EquitySeniorPerformanceFull timeTemporary workLocal areaWorldwide$184k - $356.5k
...NVIDIA is hiring senior engineers in Santa Clara, California, to develop its AI platform and enhance performance optimizations in deep learning frameworks using JAX. The role demands strong programming skills and system design expertise, along with the ability to effectively...Employment EquitySeniorPerformanceNVIDIA
Santa Clara, CA9 hours ago$152k - $241.5k
...learning ignited modern AI — the next era of... ...”.NVIDIA is hiring a Senior AI Compiler Engineer. GPUs are driving rapid... ...end, with a focus on performance, fast builds, low... ...such as PyTorch and JAX.Experience with GPU kernel... ...also be eligible for equity and benefits....Employment EquitySeniorPerformanceFull timeRemote work$184k - $287.5k
...'re looking for outstanding AI systems engineers to develop groundbreaking technologies... ...frameworks (e.g. PyTorch, JAX, TensorFlow, ONNX, etc) and... ...GPU kernel development and performance optimizations (especially... ...will also be eligible for equity and benefits.Applications...Employment EquitySeniorPerformanceFull time$151.8k - $332.2k
...re looking for an experienced AI Engineer passionate about building the foundational AI Services Platform that powers Zoom's... ...architecture, observability, and performance optimizationHold Bachelor's degree... ...consideration; base salary, bonus and equity value.Note: Starting pay will...Employment EquitySeniorPerformanceFull timeWork at officeRemote work$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... ...also be eligible for equity and benefits.Applications...Employment EquitySeniorPerformanceFull timeRemote work- ...Senior Lead Software EngineerBe an integral... ...Senior Lead Software Engineer at JPMorgan Chase... ..., Infrastructure Platforms team, you are an... ...culture of diversity, equity, inclusion, and... ...platforms optimized for AI/ML workloads.... ...resources for performance and cost efficiency...Employment EquitySeniorPerformanceFor contractors
Chase
Palo Alto, CA3 days ago $184k - $287.5k
...NVIDIA's DGX Cloud AI Efficiency Team... ...software engineer to join our team... ...AI systems.As a senior DGX Cloud AI Infrastructure... ...NVIDIA's AI platforms.Define... ...PyTorch, TensorFlow, JAX, and RayNVIDIA... ..., High-Performance Computing, and... ...be eligible for equity and benefits.Applications...Employment EquitySeniorPerformanceFull timeRemote work- ...technology products. As a Senior Lead Software Engineer at JPMorgan Chase... ..., Infrastructure Platforms team, you are an integral... ...of diversity, equity, inclusion, and respect... ...optimized for AI and machine learning... ...cloud resources for performance and cost efficiency....Employment EquitySeniorPerformanceFor contractors
J.P. Morgan
Palo Alto, CA4 days ago - ...worldwide.We’re a team of engineers, clinicians, and... ...work helps care teams perform with greater... ...next-generation robotic platforms. As a Senior AI/ML Research Engineer... ...of PyTorch/TensorFlow/JAX, and the ability to stand... ...incentives, benefits, and equity. It would not be...Employment EquitySeniorPerformanceLocal areaWorldwideFlexible hours
$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... ...5.You will also be eligible for equity and benefits.Applications for...Employment EquitySeniorPerformanceFull time$272k - $431.25k
...JAX is the framework of choice for the most ambitious AI research, and the datacenters it... ...needs to evolve to perform well across all of... .... NVIDIA seeks a Senior Engineering Manager to define... ...across the platform, from single-accelerator... ...be eligible for equity and benefits....Employment EquitySeniorPerformanceFull time$152k - $241.5k
We are now looking for a Senior Agentic AI Software Engineer! Today, NVIDIA is tapping into the unlimited... ...agentsFamiliarity with how to evaluate performance of AI agents Experience using... ...Level 4.You will also be eligible for equity and benefits.Applications for this...Employment EquitySeniorPerformanceFull time$200k - $400k
...machines at scale. At Scout AI, we’re developing Fury, the... ...We're looking for a Senior or Staff AI Engineer to join the Fury Orchestration... ...operations to validate system performance e under real-world constraints... ...and bonus Meaningful equity Premium medical, dental,...Employment EquitySeniorPerformanceFull timeRelocation packageScout Ai
Sunnyvale, CA1 day ago$152k - $241.5k
...As part of Nvidia's applied AI team for chip design, you will... ...the intersection of research, engineering, and product development,... ...their seamless and efficient performance. If you're passionate about the... ...You will also be eligible for equity and benefits.Applications for...Employment EquitySeniorPerformanceFull time$152k - $241.5k
We are seeking a Senior AI/ML Performance and Efficiency Engineer, GPU Clusters at NVIDIA to join our AI Efficiency efforts... ...familiarity with cloud computing platforms (e.g., AWS, GCP, Azure) in... ...Level 4.You will also be eligible for equity and benefits.Applications for this...Employment EquitySeniorPerformanceFull timeRemote work$184k - $287.5k
...the industry leader in high performance computing, gaming and AI. Our GPUs and SOCs give... ...alone! Now we're hiring the engineer who will lead the rebuild... ...AI to shape the world's AI platform. If that sounds exciting,... ...will also be eligible for equity and benefits.Applications...Employment EquitySeniorPerformanceFull timeImmediate start$152k - $241.5k
...are looking for a software engineer with a strong background... ...to push the limits of performance at the intersection of AI, high-performance computing... ...influence next-generation platforms, and share your insights... ...will also be eligible for equity and benefits.Applications...Employment EquitySeniorPerformanceFull time$152k - $241.5k
We are now looking for a Senior AI Frameworks Engineer (C++/Python)! NVIDIA's high-performance computing platforms are powering the AI revolution across many applications and industries... ...for Level 4.You will also be eligible for equity and benefits.Applications for this job will...Employment EquitySeniorPerformanceFull time$152k - $241.5k
...computer graphics, high-performance computing, and... ...everything from generative AI to autonomous systems,... ...scales the infrastructure, platforms, and tools that enable researchers and engineers to develop the next generation... ...also be eligible for equity and benefits....Employment EquitySeniorPerformanceFull time$152k - $241.5k
We’re currently seeking a Senior AI Developer Technology Engineer, Financial Sector!Would... ...on cutting-edge computing platforms? Is it rewarding toinvestigate... ...the best possible performance of computer hardware? Could... ...will also be eligible for equity and benefits.Applications...Employment EquitySeniorPerformanceFull timeWork experience placementRemote work$152k - $241.5k
...built in the age of Generative AI? Join NVIDIA’s TensorRT team... ...of AI agents to produce high-performance, high-quality, modern C++... ...you are a systems-thinking C++ engineer who wants to help scale out an... ...You will also be eligible for equity and benefits.Applications for...Employment EquitySeniorPerformanceFull time$244.14k - $413.16k
...integrating advanced AI and autonomous driving... ...for a hands-on Senior Staff AI Engineer to build and scale production... ..., evaluation, and performance.Job Responsibilities:... ...applications, services, and platforms leveraging state-of-... ...addition to bonus, equity and benefits. Our...Employment EquitySeniorPerformanceFull time$200k - $322k
...learning ignited modern AI — the next era of... ....Design-for-X Engineering at NVIDIA works on... ...'ll be doing:As a senior member in our team... ...while monitoring performance, automating deployments... ...with cloud platforms (AWS, Azure, GCP)... ...also be eligible for equity and benefits.Applications...Employment EquitySeniorPerformanceFull time$168k - $264.5k
...consequences: escapes, yield loss, and performance loss.We're hiring a Senior Manufacturing & System Co-Design Workflow Engineer to lead the methodology and... ...infrastructure that makes AI genuinely usable in a... ...will also be eligible for equity and benefits.Applications for...Employment EquitySeniorPerformanceFull timeImmediate start$176k - $276k
...Artificial Intelligence, High-Performance Computing and... ...our team of innovative engineers who develop and maintain... ...motivated EngOps and Platform Engineers to boost... ...also be eligible for equity and benefits.Applications... ...vacancy. NVIDIA uses AI tools in its recruiting...Employment EquitySeniorPerformanceFull time$200k - $322k
...unlimited potential of AI to define the next... ..., and hands-on Senior full-stack... ...next generation AI platforms and products that... ...productivity. This engineer is expected to be... ...reliability, safety, and performance of autonomous... ...also be eligible for equity and benefits.Applications...Employment EquitySeniorPerformanceFull timeImmediate start- NVIDIA GPU Architecture Group is seeking a senior software engineer to automate and optimize performance analysis workflows for AI training and inference workloads. You will not only perform analysis but also reshape how it's done, building tools and workflows that scale...Employment EquitySeniorPerformance
Do you want to receive more vacancies?
Subscribe and receive similar vacancies to Senior AI Platform Engineer - JAX & Performance (Equity). Be the first to apply!
Related searches
- ai developer Santa Clara, CA
- ai engineer Santa Clara, CA
- ai prompt engineer Santa Clara, CA
- senior ai engineer Santa Clara, CA
- ai engineer remote Santa Clara, CA
- platform engineer Santa Clara, CA
- platform engineering manager Santa Clara, CA
- platform developer Santa Clara, CA
- senior platform engineer Santa Clara, CA
- data platform engineer Santa Clara, CA