JAX & AI Systems Performance Intern
$20 - $71 per hourNVIDIA
NVIDIA is seeking highly motivated interns to join their team in Santa Clara, California. You will develop performance optimizations for deep learning frameworks using JAX, contributing core components to enhance the NVIDIA AI platform. Ideal candidates are pursuing a MS or PhD in Computer Science or related fields and have skills in Python, JAX, and machine learning frameworks. This role offers competitive hourly rates ranging from 20 USD to 71 USD and includes intern benefits, fostering a diverse and inclusive working environment. #J-18808-Ljbffr NVIDIA
Vacancy posted 1 day ago
Similar jobs that could be interesting for youBased on the JAX & AI Systems Performance Intern in Santa Clara, CA vacancy
$20 - $71 per hour
...unlimited potential of AI to define the next era... ...developing sophisticated performance optimizations for deep... ...learning frameworks using JAX, a powerful tool that... ...developing JAX-based systems by developing tools that... ...The hourly rate for our interns is 20 USD - 71 USD. You...InternshipPerformanceHourly payNVIDIA
Santa Clara, CA3 hours ago$30 - $94 per hour
We are now looking for a PhD Research Intern with a focus in System Software and I/O Architecture! NVIDIA... ...experience in profiling and system performance analysis tools. NVIDIA has pioneered... ...world leader in high‑performance and AI computing technology, with ambitious...InternshipPerformanceHourly pay$35 - $45 per hour
...focuses on scalable and assistive AI. We develop solutions for... ...driving, advanced driver assistance systems (ADAS), robotics, enterprise... ...for an engineering research intern. Your work will help bridge... ...maintain code quality and system performance Required: Currently pursuing...InternshipPerformanceHourly payWork experience placementLocal areaWorldwide$50 - $70 per hour
A leading AI technology firm is seeking a Data Movement Architecture Intern to work on next-generation AI and CPU platforms. The intern will collaborate with architects... ...relevant fields and possess skills in C/C++ and performance analysis. The role offers a competitive hourly...InternshipPerformanceRemote jobHourly pay$184k - $356.5k
...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...Performance- ...Phoenix, Inc. Presto is the leading Voice AI company for restaurant drive‑thrus,... ...and are excited to work on real‑world AI systems alongside a high‑performance team. The Role We are looking for Fall 2026 AI Engineering Interns to contribute to the core intelligence behind...InternshipPerformanceFull time
$184k - $287.5k
...engineers to develop its AI platform and more specifically its performance optimizations in deep learning frameworks using JAX, a tool that can differentiate... ...skills and excellent system design skills. You should... ...frameworks and their internals (e.g. PyTorch, TensorFlow...Performance$184k - $287.5k
...of CUDA and Deep Learning Systems. As the complexity and scale... ...help unlock maximum hardware performance for emerging AI workloads. You will be a... ...upstream framework integrations, internal tools, or closed-source... ...frameworks (e.g., PyTorch, JAX, TensorRT, vLLM, sgLang,...Performance$180k - $250k
Member of Technical Staff -- TPU Systems (JAX / XLA / PALLAS) About the... ...Systems Engineer to build high-performance inference and training... ...Deep understanding of JAX/XLA internals: HLO, fusion, SPMD partitioning... ...engineers who've shipped production AI systems, created SGLang (20K+...PerformanceFull timeFlexible hours$184k - $356.5k
NVIDIA Gruppe is seeking senior engineers to enhance its AI platform, focusing on performance optimizations in deep learning frameworks using JAX. Candidates should have strong programming and system design skills, along with proficiency in C/C++ and Python. The role involves...Performance$80k - $85k
...maintain the highest quality and performance standards in everything we do.... ...Software Test Engineers (aka System Test engineers) to help us in... ...architecture and leverages ML/AI to simplify operations,... ...desirable Compensation: The intern base pay for this role is$80,0...InternshipPerformanceNight shift3 days per week- ...positions, not internships. International Students: If your work authorization... ...up RTOS (Real Time Operating Systems) on the next generation of... ...Linux device drivers for AI interface hardware accelerators... ...data to improve design and performance of a system Understanding of...InternshipPerformanceFull timeTemporary workPart timeFlexible hours
- ...the infrastructure behind the AI-driven data economy. As AI... ...interaction, every model, every system generates data that must be... ...analytical, and enthusiastic IT Intern for Wafer systems Automation Eco... ...career - learning mindset, performance goals and feedback, personal career...InternshipPerformanceWork experience placementWork at officeWorldwide
$224k - $356.5k
...is a workstation-class AI computer built on GB300... ...(PyTorch, TensorFlow, JAX) for training and inference... .... Characterize performance across model sizes, batch... ...related field. 12+ years in systems software engineering... ...TensorFlow, or JAX—including internals: graph execution,...Performance$19 - $65 per hour
...Opportunity PlusAI is a Physical AI company pioneering AI-based... ...GROUP's Scania, MAN, and International brands, Hyundai Motor Company... ...: Design and implement high-performance custom compute kernels using... ...Automate Profiling Workflows: Build systems to automate performance...InternshipPerformanceHourly payPlusAI, Inc.
Santa Clara, CA2 days ago- ...Monolithic Power Systems, Inc. MonolithicPowerSystems,Inc.(MPS)isoneofthefastestgrowingcompaniesintheSemiconductorindustry... ...Summary We are seeking a highly skilled AI Developer to join our innovative team.... ..., Deepseek, and Nvidia to optimize AI performance. Collaborate with Cross-Functional Teams:...InternshipPerformanceHourly pay
Dormont Manufacturing Company
San Jose, CA4 days ago - ...innovation, integrating advanced AI and autonomous driving... ...into real‑time, production‑ready systems on vehicle hardware remains a... ...for an entry‑level engineer or intern to support the optimization and... ...driving systems. Help analyze model performance, memory usage, latency, and...InternshipPerformance
XPENG
Santa Clara, CA1 day ago - PlusAI, a Silicon Valley-based Physical AI company, seeks engineers to optimize image... ...features. You will work on scalable backend systems, integrating ML concepts, embeddings, and... ...databases to improve relevance and performance. The role involves cross-domain collaboration...InternshipPerformance
$50 - $70 per hour
...Physical Design Intern - CPU/AI Hardware Tenstorrent is leading the industry on cutting‑edge AI technology, revolutionizing performance expectations, ease of use, and cost efficiency. With... ...or indirect access to information, systems, or technologies subject to these laws...InternshipPerformancePermanent employmentAustin Community College
Santa Clara, CA4 days ago$272k - $431.25k
JAX is the framework of choice for the most ambitious AI research, and the datacenters it runs on are evolving fast. NVIDIA... ..., and JAX needs to evolve to perform well across all of them. NVIDIA... ...prototype, and build innovative systems that improve products and processes...Performance- ...and product innovations through the use of AI by making it trusted, safe and ubiquitous... ...scalable, high-availability backend systems. Work with the development team to integrate... ...functional teams to identify and resolve performance bottlenecks. Stay updated on new technologies...InternshipPerformance
brevian-careers
Sunnyvale, CA3 hours ago - ...innovation, integrating advanced AI and autonomous driving... ...development work. We’re looking for an intern to help build the data... ...experiment data to improve agent performance on domain‑specific tasks. Key... ...development, LLM evaluation, agentic AI systems, RAG pipelines, semantic...InternshipPerformance
XPENG
Santa Clara, CA3 days ago $19 - $65 per hour
We're seeking a top Research Intern to join our team focusing on perception situational... ...challenges in long-range perception to enhance system performance, ensuring safety-critical standards are... ...and validation of safety-critical AI systems. You'll design and implement cutting...InternshipPerformanceHourly pay$50 - $70 per hour
...leading the industry on cutting‑edge AI technology, revolutionizing performance expectations, ease of use, and cost... ..., firmware, software and system engineering teams. This role is on‑... ...analytical skills Compensation for all interns at Tenstorrent ranges from $50/hr -...InternshipPerformancePermanent employmentTenstorrentuniversity
Santa Clara, CA3 days ago$200k
...talented and self-motivated Physical AI Engineer (Robotics & Agentic Systems) to join it in advancing the... ...System Optimization: Implement the JAX AI Stack (JAX, Flax, Optax) for efficient... ...This role may also be eligible for performance-based bonuses subject to company...PerformanceLocal area$20 - $71 per hour
...looking for Java engineering interns to work on cuVS, a suite of open... ..., but exposes that high-performance GPU compute through user-friendly... ...of data processing and AI. You’ll work closely with the... ...algorithms and running on distributed systems: HPC, Cloud, etc Distributed...InternshipPerformanceHourly pay- Baidu USA is seeking a Machine Learning System Software Architect in Sunnyvale, CA. You'll architect high-performance machine learning software solutions and lead the creation of distributed AI systems. The ideal candidate has 5+ years of experience, deep knowledge of ML...Performance
- PlusAI is a Physical AI company pioneering AI-based virtual driver... ...GROUP’s Scania, MAN, and International brands, Hyundai Motor Company... ...use cases, benchmark their performance against key workload requirements... ..., algorithms, operating systems, and file I/O performance. Experience...InternshipPerformance
$50 - $55 per hour
...video‑processing workflows for internal teams and external partners.... ...reliable video‑processing performance. Implement and improve low‑level... ...Explore opportunities to use AI‑ and LLM‑based techniques to... ...C++ fundamentals, including system‑level programming, memory management...InternshipPerformanceHourly payFull timeWork experience placementWorldwideMonday to FridayFlexible hoursDolby Laboratories
Sunnyvale, CA3 hours ago- Baidu US, located in Sunnyvale, California, is seeking a Machine Learning System Software Architect to develop high-performance system software for AI. The ideal candidate will have over 5 years of experience in architecting software solutions for machine learning and...Performance
Do you want to receive more vacancies?
Subscribe and receive similar vacancies to JAX & AI Systems Performance Intern. Be the first to apply!
Related searches
- application performance engineer Santa Clara, CA
- acting performance Santa Clara, CA
- performance test architect Santa Clara, CA
- lead performance test engineer Santa Clara, CA
- performance test engineer Santa Clara, CA
- performance engineer Santa Clara, CA
- senior performance engineer Santa Clara, CA
- senior performance tester Santa Clara, CA
- sports performance Santa Clara, CA
- performance testing Santa Clara, CA