NVIDIA · 10 hours ago
Senior Software Engineer, ML
Santa Clara, CA
Full-time
Onsite
Senior Level, Lead/Staff
$168K/yr - $264K/yr
8+ years expNVIDIA is a leading company in computer graphics and AI technology, seeking senior engineers focused on performance analysis and optimization in machine learning. The role involves implementing and optimizing ML models, collaborating with researchers, and influencing the hardware and software roadmap in the AI domain.
Artificial Intelligence (AI)Consumer ElectronicsGPUHardwareSoftwareVirtual Reality
Responsibilities
Implement machine learning models from multiple data domains (CV, NLP/LLMs, ASR, TTS, RecSys and others) in multiple ML frameworks (PyT, JAX, TF2, DGL and others)
Implement and test new SW features (Graph Compilation, reduced precision training) that use the most recent HW functionalities
Analyze, profile, and optimize machine learning workloads on pioneering hardware and software platforms
Collaborate with researchers and engineers across NVIDIA, providing guidance on improving the design, usability and performance of models
Lead best-practices for building, testing, and releasing ML software
Qualification
Required
8+ years of experience in ML model implementation and SW Development
BSc, MS or PhD degree in Computer Science, Computer Architecture, Mathematics, Physics or related technical field or equivalent experience
Excellent Python programming skills, extensive knowledge of at least one ML Framework
Strong problem solving and analytical skills
Algorithms and ML fundamentals
Experience with Git or other source-control systems
Published papers, patents or articles
Benefits
Equity
Benefits
Company
NVIDIA
NVIDIA is a computing platform company operating at the intersection of graphics, HPC, and AI.
10001+ employees
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
Represents job field similar to this job
Trends of Total Sponsorships
2025 (1877)
2024 (1355)
2023 (976)
2022 (835)
2021 (601)
2020 (529)
Funding
Current Stage
Public CompanyTotal Funding
$4.09BKey Investors
ARPA-EARK Investment ManagementSoftBank Vision Fund
2023-05-09Grant· $5M
2022-08-09Post Ipo Equity· $65M
2021-02-18Post Ipo Equity
Leadership Team
Jensen Huang
Founder and CEO
Michael Kagan
Chief Technology Officer
Recent News
2026-01-01
2026-01-01
Business Insider
2026-01-01
Company data provided by crunchbase