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 leading technology company in High-Performance Computing, Artificial Intelligence, and Visualization, is seeking a Machine Learning Software Platform Architect. 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 chip design applications. The position offers an opportunity to work with cutting-edge technology, collaborating with hardware designers and LLM research teams to build and optimize AI-powered solutions for GPU design processes.

The role involves developing LLM-based applications such as QA bots and code generators, managing training data for hardware-specific language models, and ensuring robust infrastructure performance. The ideal candidate will bring strong expertise in Python, web development, and LLM-related technologies, combined with an understanding of chip design challenges. This position offers competitive compensation, including a base salary range of $148,000 - $356,500, plus equity and comprehensive benefits.

NVIDIA's commitment to innovation and its position at the forefront of AI and GPU technology makes this an exceptional opportunity for someone passionate about applying machine learning to solve complex hardware design challenges. The company's culture promotes creativity, forward-thinking, and collaboration, making it an ideal environment for professionals looking to make significant contributions to groundbreaking technological developments.

Last updated a day ago

Responsibilities For Machine Learning Software Platform Architect

  • Develop and maintain infrastructure for managing 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 or equivalent experience
  • 5+ years experience
  • Experience in developing and maintaining AI/ML infrastructure
  • 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

Equity
  • Equity
  • Competitive Benefits Package

Interested in this job?

Jobs Related To NVIDIA Machine Learning Software Platform Architect

Senior Deep Learning Performance Architect

Senior Deep Learning Performance Architect role at NVIDIA focusing on developing advanced processor architectures for AI acceleration, offering competitive compensation and the chance to shape the future of machine learning.

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 developing LLM infrastructure for chip design applications

Staff Machine Learning Engineer - ML Algorithms

Staff Machine Learning Engineer position at EarnIn, focusing on developing and deploying advanced ML solutions and LLMs for fintech applications, offering competitive compensation and hybrid work arrangement.