Machine Learning Software Platform Architect

NVIDIA is the world leader in accelerated computing, pioneering GPU technology and transforming industries through AI and digital twins.
Machine Learning
Staff Software Engineer
Hybrid
5,000+ Employees
5+ years of experience
AI · Enterprise SaaS

Description For Machine Learning Software Platform Architect

NVIDIA, a global leader in accelerated computing and AI technology, is seeking a Machine Learning Software Platform Architect to join their innovative team. This role sits at the crucial intersection of GPU chip design and artificial intelligence, focusing on developing and maintaining infrastructure for large language models (LLMs) specifically tailored for hardware design applications.

The position offers a unique opportunity to work with cutting-edge technology at one of the most respected companies in the tech industry. As a Platform Architect, you'll be responsible for building and optimizing LLM-based applications that serve hardware engineers, including QA bots and code generators. You'll collaborate closely with both hardware chip designers and LLM research teams, bridging the gap between advanced AI capabilities and practical hardware design challenges.

The ideal candidate brings 5+ years of experience in AI/ML infrastructure development, strong Python programming skills, and deep understanding of LLM technologies including langchain, vector databases, and prompt engineering. Your role will be essential in advancing NVIDIA's capabilities in using AI to enhance chip design processes.

Working at NVIDIA means joining a company at the forefront of AI and GPU technology, with competitive compensation and comprehensive benefits. You'll be part of a forward-thinking team that's pushing the boundaries of what's possible in AI and hardware design, making this an excellent opportunity for someone passionate about both machine learning and high-performance computing.

Last updated 3 hours ago

Responsibilities For Machine Learning Software Platform Architect

  • Develop and maintain infrastructure for managing LLMs specifically adapted for chip design and hardware domain
  • Develop and maintain LLM based applications like QA bot and code generator
  • Collaborate with hardware chip designers and LLM research teams
  • Collect & organize training/fine-tuning data for hardware specific language models
  • Optimize infrastructure for performance, scalability, and reliability
  • Ensure secure and efficient data management

Requirements For Machine Learning Software Platform Architect

Python
  • 5+ years work experience in AI or machine learning infrastructure
  • BS in computer science or related field
  • Strong proficiency in Python and web development
  • Knowledge of LLM techniques (langchain, vector database, prompt engineering)
  • Understanding of chip design and related challenges
  • Experience with data management
  • In-depth understanding of Machine Learning/Deep Learning/NLP concepts

Benefits For Machine Learning Software Platform Architect

Medical Insurance
  • Competitive salaries
  • Comprehensive benefits package

Interested in this job?

Jobs Related To NVIDIA Machine Learning Software Platform Architect

Deep Learning Engineer - Distributed Task-Based Backends

Senior to Principal Deep Learning Engineer position at NVIDIA focusing on distributed backends for AI frameworks, offering remote work and competitive compensation.

Senior Computer Architect - Deep Learning

Senior Computer Architect position at NVIDIA focusing on deep learning architecture design for next-generation GPUs and AI acceleration.

Machine Learning Software Platform Architect

Senior ML Platform Architect role at NVIDIA focusing on LLM infrastructure for chip design

Manager, Deep Learning Algorithms

Lead engineering activities for Deep Learning models at NVIDIA, managing teams and projects in AI and accelerated computing.

Deep Learning Engineer - Distributed Task-Based Backends

Senior to Principal Deep Learning Engineer position at NVIDIA focusing on distributed backends for AI frameworks, offering remote work and competitive compensation.