54 applicants

Company

Original Job Post

NVIDIA · 3 days ago

Machine Learning Software Platform Architect

Santa Clara, CA

Full-time

Onsite

Senior Level

$148K/yr - $339K/yr

5+ years exp

Wonder how qualified you are to the job?

Maximize your interview chances

Artificial Intelligence (AI)GPU

H1B Sponsorship

Growth Opportunities

Hiring Manager

Sarmita Chatterjee

Insider Connection @NVIDIA

Discover valuable connections within the company who might provide insights and potential referrals, giving your job application an inside edge.

Responsibilities

Develop and maintain the infrastructure for managing large language models (LLMs) based applications specifically adapted for chip design and hardware domain.

Develop and maintain LLM based applications to serve hardware engineers, such as LLM based QA bot, code generator, etc.

Collaborate with HW chip designers and LLM research teams to understand the specific needs and challenges of GPU design and ensure the LLM infrastructure is well-suited to these needs.

Collaborate with LLM research teams to collect & organize training / fine-tuning data to train hardware specific language model.

Optimize the infrastructure for performance, scalability, and reliability, and ensure the secure and efficient management of data.

Stay updated with the latest industry trends in AI and machine learning, and continuously look for opportunities to apply these advancements to improve the LLM infrastructure.

Qualification

Find out how your skills align with this job's requirements. If anything seems off, you can easily click on the tags to select or unselect skills to reflect your actual expertise.

PythonWeb DevelopmentAIMachine LearningInfrastructureLarge Language ModelsLangchainVector DatabasePrompt EngineeringChip DesignData ManagementDoc CleaningSecure StorageMachine Learning ConceptsDeep Learning ConceptsNLP ConceptsProblem-SolvingTeamworkMicroservicesCloud/Distributed InfrastructureFront-End DevelopmentReactVue.jsSQLNoSQL

Required

BS in computer science or related or equivalent experience

5+ years experience

Experience in developing and maintaining AI or machine learning infrastructure, preferably in the context of large language models

Strong proficiency in Python and web development, and familiarity with LLM related techniques e.g., langchain, vector database, prompt engineering, etc.

Understanding of chip design and related computational and data challenges

Experience with data management, including doc cleaning, transformation, and secure storage

Excellent problem-solving skills and the ability to work effectively in a team

In-depth understanding of Machine Learning / Deep Learning / NLP concepts

Preferred

You crafted & developed production quality microservices

Strong technical background in cloud/distributed infrastructure

An excellent plus if you are familiar with front-end development using React or Vue.js

Strong understanding of SQL & NoSQL Data platforms

Benefits

Equity

Company

NVIDIA

Glassdoor

4.6

NVIDIA is a computing platform company operating at the intersection of graphics, HPC, and AI.

Founded in 1993

Santa Clara, California, USA

10,001+ employees

https://www.nvidia.com

H1B Sponsorship

NVIDIA has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)

Distribution of Different Job Fields Receiving Sponsorship

Trends of Total Sponsorships

2023 (735)

2022 (892)

2021 (696)

2020 (534)