Reddit is a community of communities. It’s built on shared interests, passion, and trust, and is home to the most open and authentic conversations on the internet. Every day, Reddit users submit, vote, and comment on the topics they care most about. With 100,000+ active communities and approximately 130 million daily active unique visitors, Reddit is one of the internet’s largest sources of information. For more information, visit www.redditinc.com . Who We Are: The Machine Learning Platform team at Reddit is a high-impact team that owns the infrastructure that powers recommendations, content discovery, user and content quantification, while directly impacting other teams such as Growth, Ads, Feeds, and Core Machine Learning teams. What You’ll Do: As a Senior ML Infrastructure Engineer, you will lead development of a platform for large scale ML models at Reddit. Design end-to-end model lifecycle patterns (MLOps) to boost velocity of development for ML engineers, including data preparation, model management, experiment tracking, and more Zero-to-one development and support of a graph ML codebase and platform that abstracts away common patterns and enables greater model scalability and iteration Collaborate with ML engineers on performance tuning, including improving model training time, efficiency, and GPU training costs in a large, distributed ML training environment Optimize batch data processing within a data warehouse and with tools such as Apache Beam, Apache Spark, Ray Data, and more Architect pipelines to build and maintain massive graph data structures on the order of billions of nodes and tens of billions of edges Who You Might Be: 5+ years of experience in ML infrastructure, including model training and model deployments Hands-on experience with ML optimization, including memory and GPU profiling Deep experience with cloud-based technologies for supporting an ML platform, including tools like GCP BigQuery, Google Cloud Storage, infrastructure-as-code
Senior GPU System Architect, Context Management and Register Architecture
Salary not disclosed
Check market pay for comparable Senior GPU System Architect roles before applying.
Role overview
Job description
NVIDIA is seeking a Senior System Architect to define two foundational areas of the GPU: the mechanisms that manage GPU execution state and the register address and access architecture used across the processor. This hands-on role spans architecture, functional modeling, verification, hardware and software integration, and silicon bring-up. We collaborate across ASIC, firmware, driver, verification, and performance teams to guide features from early exploration through implementation and validation. The resulting architecture will build GPU responsiveness, reliability, scalability, software compatibility, and the way increasingly sophisticated processors are configured and managed.
What you’ll be doing:
- Architect and evolve GPU context-management mechanisms, including efficient methods for saving and restoring execution state.
- Own the GPU-wide register address map, coordinate address allocation, resolve conflicts, and maintain a scalable architecture through multiple product iterations.
- Define requirements, interfaces, compatibility expectations, and on-die and cross-die interconnects for register access and programming.
- Develop C++ functional-model support and collaborate with hardware build, embedded software, device software, testing, and performance groups to translate architectural requirements into coordinated solutions.
- Evaluate tradeoffs across performance, area, complexity, scalability, reliability, compatibility, and development effort.
- Build validation strategies, debug cross-stack issues through pre-silicon validation and initial silicon testing, and improve the tools and automation supporting register-interface development.
What we need to see:
- A bachelor’s degree in Computer Engineering, Computer Science, Electrical Engineering, or a similar field, or equivalent experience.
- 5+ years of proven experience in computer architecture, GPU architecture, CPU architecture, system-on-chip design, or a related area.
- Experience designing, modeling, or validating sophisticated hardware features from definition through confirmation, with proficiency in C++ and Python or a comparable scripting language.
- Strong understanding of interactions among hardware, firmware, and system software, with a demonstrated ability to diagnose and resolve issues spanning components or abstraction layers.
- Experience developing coherent architectural standards across multiple teams, paired with clear technical writing skills and experience crafting specifications and collaborating with distributed, cross-functional groups.
Ways to stand out from the crowd:
- Experience with hardware register architecture, address-map development, memory-mapped I/O, or device-control interfaces.
- Knowledge of context switching, scheduling, virtualization, or execution-state management.
- Familiarity with RISC-V or other embedded-controller architectures, on-chip interconnects, or cross-die communication fabrics.
- Experience with firmware, device drivers, functional or performance modeling, hardware verification, or test development.
- Hands-on experience with pre-silicon validation, silicon bring-up, automated interface generation, or maintaining architectural compatibility across product generations.
#ArchitectYourFuture
Your base salary will be determined based on your location, experience, and the pay of employees in similar positions. The base salary range is 184,000 USD - 287,500 USD for Level 4, and 224,000 USD - 356,500 USD for Level 5.You will also be eligible for equity and benefits.
This posting is for an existing vacancy.
NVIDIA uses AI tools in its recruiting processes.
NVIDIA is committed to fostering an inclusive work environment and proud to be an equal opportunity employer. As we highly value diversity in our current and future employees, we do not discriminate (including in our hiring and promotion practices) on the basis of race, religion, color, national origin, gender, gender expression, sexual orientation, age, marital status, veteran status, disability status or any other characteristic protected by law.What they are looking for
Skills & requirements
Qualification
What we need to see: A bachelor’s degree in Computer Engineering, Computer Science, Electrical Engineering, or a similar field, or equivalent experience
Hiring company
NVIDIA
We create the world’s fastest supercomputer and largest gaming platform.
Related opportunities
Similar active roles
Relevant, currently active roles for this market.
$243.3K – $295.3K/yr
🔔 Get job alerts
New Senior GPU System Architect, Context Management and Register Architecture jobs in Ca, Santa Clara, United States, straight to your inbox.
No spam · Unsubscribe anytime