Machine Learning Software Platform Architect

World leader in accelerated computing, pioneering AI and digital twins technology to transform industries.
$148,000 - $356,500
Machine Learning
Staff Software Engineer
In-Person
5,000+ Employees
5+ years of experience
AI · Enterprise SaaS

Description For Machine Learning Software Platform Architect

NVIDIA, a pioneer in GPU technology and artificial intelligence, 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 AI, focusing on developing and maintaining infrastructure for large language models (LLMs) specifically adapted for the hardware domain.

The position offers an exciting opportunity to work with cutting-edge technology in AI and chip design, combining both fields to create innovative solutions. You'll be responsible for developing LLM-based applications that serve hardware engineers, including QA bots and code generators, while collaborating closely with hardware chip designers and LLM research teams.

The ideal candidate will bring strong expertise in Python, web development, and LLM-related techniques, along with a deep understanding of machine learning concepts. You'll be working in Santa Clara, CA, with a competitive salary range of $148,000 to $356,500, plus equity and comprehensive benefits.

NVIDIA's position as an industry leader in High-Performance Computing, AI, and Visualization makes this an exceptional opportunity for someone passionate about applying machine learning to solve complex problems in hardware design. The role offers the chance to work with some of the most forward-thinking professionals in the industry while contributing to groundbreaking developments in AI and chip design.

The company's commitment to innovation, coupled with its strong market position and continuous growth, makes this an ideal opportunity for a seasoned professional looking to make a significant impact in the intersection of AI and hardware technology. You'll be part of a team that's pushing the boundaries of what's possible in AI and chip design, while enjoying the benefits of working for one of the technology world's most desirable employers.

Last updated a day ago

Responsibilities For Machine Learning Software Platform Architect

  • Develop and maintain infrastructure for managing large language models (LLMs) for chip design domain
  • Develop and maintain LLM based applications like QA bot and code generator
  • Collaborate with HW 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
  • BS in computer science or related field
  • 5+ years experience
  • Experience in AI/ML infrastructure development
  • 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 ML/DL/NLP concepts

Benefits For Machine Learning Software Platform Architect

Medical Insurance
Equity
  • Competitive salaries
  • Comprehensive benefits package
  • Equity

Interested in this job?

Jobs Related To NVIDIA Machine Learning Software Platform Architect

Software Manager, Deep Learning Frameworks

Lead NVIDIA's Deep Learning Frameworks team, developing core ML software and algorithms while managing engineering talent.

Gen AI Product Evangelist Engineer, Retail

Senior technical role combining AI expertise with developer advocacy to create educational content and demos for NVIDIA's retail-focused Gen AI applications.

Manager, Neural Reconstruction Software

Lead NVIDIA's Neural Reconstruction Software team, developing cutting-edge AI technologies for computer vision and robotics applications.

Senior Manager, Visual AI Microservices

Lead the development of Visual AI Microservices at NVIDIA, managing teams building cutting-edge computer vision and AI services for developer enablement.

Manager, Artificial Intelligence Algorithms

Lead AI algorithms team at NVIDIA, managing deep learning software development and optimization for GPU-accelerated platforms.