AI Systems Intern — JAX Performance and ML Tools
NVIDIA Gruppe
NVIDIA Gruppe is seeking interns to develop performance optimizations for JAX, enhancing deep learning frameworks. Responsibilities include designing core components and improving team efficiency through tool development. The ideal candidate is pursuing a MS or PhD in related fields, has a strong grasp on JAX and machine learning frameworks, and is proficient in Python. A diverse and inclusive work environment is promised. #J-18808-Ljbffr NVIDIA Gruppe
Vacancy posted 5 days ago
Similar jobs that could be interesting for youBased on the AI Systems Intern — JAX Performance and ML Tools in Santa Clara, CA vacancy
$20 - $71 per hour
NVIDIA Corporation is seeking passionate interns to contribute to JAX development for AI and deep learning optimization. This role requires strong programming skills in Python, with familiarity in C++. Candidates should be pursuing a MS or PhD in Computer Science or Engineering...InternshipHourly pay- ...Design and implement agentic AI workflows , including planning, reasoning, tool usage, and memory... ...in building multiagent systems and agent orchestration... ...Evaluate agent behavior, performance, reliability, and failure... ...Understanding of Basic ML and deep learning concepts...InternshipPerformance
$6,710 per month
...Artificial Intelligence (AI) increasingly... ...on breakthroughs in systems and architecture,... ...AI systems through performance modeling, architectural... .... Research Interns will investigate topics... ...computer architecture, AI/ML systems,... ...performance analysis tools. Familiarity with...InternshipPerformanceOngoing contractSummer workLocal areaMicrosoft Corporation
Mountain View, CA3 days ago$44k - $185k
...the future of Cisco's AI-driven platforms and data... ...data and intelligent systems. Explore the opportunities... ...software tools and platforms, you enable... ...-on experience with AI/ML. ~ Familiarity... ...Employees on sales plans earn performance-based incentive pay on...InternshipPerformanceFull timeTemporary workApprenticeshipLocal areaFlexible hoursCisco
San Jose, CA1 day ago$224k - $356.5k
...workstation-class AI computer—built... ...deeply technical systems software... ...delivers best-in-class performance for real AI workloads... ..., TensorFlow, JAX) across training... ...in AI/ML workload optimization... ...or JAX—including internals: graph execution... ...NVIDIA uses AI tools in its recruiting...PerformanceLocal areaNVIDIA
Santa Clara, CA1 day ago$25 - $70 per hour
...in Computer Science, Machine Learning, AI, or Data Science. Interns will learn to build and maintain data pipelines, optimize backend systems, and work on AI applications using... ...potential for a full-time position based on performance. Compensation ranges from $25 to $70/...InternshipPerformanceFull time- ...to join our growing AI and Generative AI... ...of large-scale AI systems powering next-generation... ..., safety, and performance of production AI services... ...for large-scale ML training, inference... ...testing, and tool-calling reliability... ...PyTorch, TensorFlow, JAX, or DeepSpeed. Experience...Performance
$50 - $70 per hour
...Learning for Physical Design Intern - CPU/AI Hardware Austin,... ..., revolutionizing performance expectations, ease of... ...into industry leading AI/ML architectures. The... ...will develop ML-based tools and flows to improve the... ...How to apply ML systems and flows to high-performance...InternshipPerformanceTenstorrent University
Santa Clara, CA4 days ago$20 - $71 per hour
...unlimited potential of AI to define the next... ...developing sophisticated performance optimizations for... ...learning frameworks using JAX, a powerful tool that combines... ...developing JAX-based systems by develop tools that... ...hourly rate for our interns is 20 USD - 71 USD....InternshipPerformanceHourly payNVIDIA
Santa Clara, CA5 days ago$40 - $50 per hour
...Agentic AI Research Intern Job Location: Santa Clara, California... ..., Data Science, System Components and Platform... ...trajectories, improving agent performance with training and test... ...that can plan, use tools, inspect state,... ...Publication record in top AI/ML, software engineering,...InternshipPerformanceVisa sponsorshipRelocation packageFujitsu
Santa Clara, CA2 days ago$45 per hour
...work on improving the performance and efficiency of large-scale AI models across... ...in high-performance ML systems and contribute to enhancing... ...and profiling tools to identify... ...PyTorch, TensorFlow, JAX). - Knowledge of fundamental... ...(Hourly) - Campus Intern The hourly rate...InternshipPerformanceHourly payFull timeSummer workLocal areaTik Tok
San Jose, CA1 day ago$160k - $183k
...AI Vision Processors For Edge Applications Our... ...for excellence in video performance by offering pristine imagery... ...welfare benefits! System Software Algorithm engineers... ...hardware and software tool stack, engineers develop... ...Develop ML and computer vision algorithms...PerformanceFull time3 days per weekAmbarella
Santa Clara, CA4 days ago$124k - $195.5k
...at the cutting edge of AI infrastructure. As agentic... ...reason about complex system-level traffic at scale.... ...you have a passion for performance analysis, a strong... ...characterizing or benchmarking ML inference workloads... ...026. NVIDIA uses AI tools in its recruiting processes...PerformanceNVIDIA
Santa Clara, CA11 hours ago- ...AI Agent Data Pipeline Intern Santa Clara, CA XPENG is a leading... ...data to improve agent performance on domain-specific tasks... ...to internal tools, dashboards, or reports... ...evaluation, agentic AI systems, RAG pipelines, semantic... ...and collaborating with ML and platform...InternshipPerformance
XPENG
Santa Clara, CA1 day ago $176k - $420k
...Optimus team, you will build the tools and infrastructure to make and... ...and accelerating the entire ML lifecycle. We are seeking an... ...robotic intelligence at scale. The systems you create will drive... ...visualizing model metrics and performance ~ Manage, analyze, and...PerformanceHourly payFull timeTemporary workFlexible hoursTesla
Palo Alto, CA3 days ago$184k - $287.5k
...sophisticated analysis / debugging tools that empower NVIDIA... ...developing tools for AI researchers and SW/HW... ...the workload and the system, and empower them to... ...issues to help improve the performance and efficiency of the... ...’ll be doing: Build internal profiling and analysis...PerformanceNVIDIA
Santa Clara, CA4 days ago- ...Principal AI/ML System Software Engineer At d-Matrix, we are focused on unleashing the potential of generative... ...Linux environment and using standard development tools Experience with distributed, high-performance software design and implementation Self-motivated...PerformanceWork experience placement3 days per week
d-Matrix
Santa Clara, CA4 days ago - ...computing experiences-from AI and data centers, to PCs, gaming and embedded systems. Grounded in a culture... ...about improving the performance of key applications and... ...with a special focus on AI/ML workloads and GPU-accelerated... ...performance-analysis tools for GPU optimization and...Performance
Advanced Micro Devices , Inc.
Santa Clara, CA1 day ago - A leading AI technology company located in Sunnyvale, California, is looking for an experienced... .... Responsibilities include bringing ML models to life on Cerebras CSX systems, performance tuning, and contributing to tool improvements. This role offers a competitive...Performance
$224k - $356.5k
...unlimited potential of AI to define the next era... ...forefront of AI and high-performance computing. As a Senior... ...Model Evaluation & AI Systems, you will play a meaningful... ..., libraries, or tools used by other engineers... ...frameworks, benchmarks, or ML infrastructure used by...PerformanceNVIDIA
Santa Clara, CA5 days ago$254k - $349.25k
...how people, data, and AI agents connect across email... ..., and collaboration tools. Over 80 of the Fortune... ...are seeking a Principal ML Architect to lead the... ...of next-generation AI systems for cybersecurity ,... ...continuously improve model performance and reliability...PerformanceFlexible hoursProofpoint
Sunnyvale, CA3 days ago$184k - $287.5k
...engineers to join us and build AI inference systems that serve large-scale... ...architect and implement high-performance inference stacks, optimize GPU... ...benchmarking methodologies and tools; contribute both new... ...pareto frontier for the field of ML Systems; survey recent publications...PerformanceNVIDIA
Santa Clara, CA5 days ago- ...continuously improve AI-powered tools and automations across... ...and decision‑support systems. Process Automation: Identify... ...automations using AI/ML, RPA, or integration... ...on user feedback and performance metrics. Technical... ...tools built, enabling internal teams to maintain and...PerformanceTemporary workFor contractors
Central Business Solutions
Santa Clara, CA13 hours ago $136k - $218.5k
...advanced analytics and AI, including LLMs... ...‑level power using internal and industry‑standard... ...gate‑level power tools, and translate data... ...and flows, including ML/RL‑based techniques... ...for power Perform comparative power... ..., architecture, or system‑level optimization....PerformanceNVIDIA
Santa Clara, CA1 day ago- ...or Hybrid) About Velaura AI Velaura AI is building the... ...beyond the cloud into edge systems and physical environments... ...design to deliver breakthrough performance per watt across cloud,... .... Experience working with ML frameworks such as PyTorch, JAX, or TensorFlow. Ability to...Performance
$254k - $349.25k
...how people, data, and AI agents connect across email... ..., and collaboration tools. Over 80 of the Fortune... ...are seeking a Principal ML Architect to lead the... ...of next-generation AI systems for cybersecurity , leveraging... ...improve model performance and reliability Productionization...PerformanceFlexible hours- Applied AI Machine Learning Lead Lead the development... ..., recommendation systems, information retrieval,... ...Augmented Generation (RAG), tool/function‑calling agents... ...Select and innovate on ML strategies for various... ...evaluate the ongoing performance of developed ML systems...Performance
- ...Hybrid) About Velaura AI Velaura AI is building... ...beyond the cloud into edge systems and physical... ...to deliver breakthrough performance per watt for AI workloads... ...Collaborate with software and ML teams to incorporate... ...or performance modeling tools.Experience with memory...Performance
- ...experience - enabling data and AI workloads to run... ..., and obsessing over performance and DevOps - this is... ...will build the "nervous system" of our AI stack-... ...Developer Velocity: Build internal tooling, SDKs, and "AI... ...+ years focused on AI/ML systems. Expert proficiency...PerformanceWork from homeFlexible hours
Cloudera
Alviso, CA1 day ago $131k - $175k
...Senior Hardware Systems Engineer – AI Rack & Cluster Infrastructure... ...of quality and performance in everything we do.... ...directly with customers and internal teams to bring... ...hyperscalers or large-scale AI/ML cluster deployments... ...automation tools for BOM management, configuration...PerformanceRemote workFlexible hoursArista Networks, Inc.
Santa Clara, CA1 day ago
Do you want to receive more vacancies?
Subscribe and receive similar vacancies to AI Systems Intern — JAX Performance and ML Tools. Be the first to apply!
Related searches
- acting performance Santa Clara, CA
- lead performance test engineer Santa Clara, CA
- performance engineer Santa Clara, CA
- high performance computing engineer Santa Clara, CA
- performance food group Santa Clara, CA
- performance test engineer Santa Clara, CA
- senior performance tester Santa Clara, CA
- IT performance management Santa Clara, CA
- performance testing Santa Clara, CA
- application performance engineer Santa Clara, CA