NVIDIA · 3 hours ago
Senior Generative AI Software Engineer
Santa Clara, CA
Full-time
Onsite
Senior Level, Lead/Staff
$224K/yr - $356K/yr
10+ years expNVIDIA is a leading company in AI technology, and they are looking for a Senior Generative AI Software Engineer to join their Cosmos generative AI engineering team. The role involves owning and evolving codebases, integrating models, and ensuring code quality and maintainability in the context of generative modeling and AI computing.
Artificial Intelligence (AI)Consumer ElectronicsGPUHardwareSoftwareVirtual Reality
Responsibilities
You will own and evolve the Cosmos open-source and internal research codebases, crafting core infrastructure that supports our foundation model research and deployment
Refactor and modularize large research-driven code into clean, testable, maintainable libraries for use across teams
Integrate and adapt off-the-shelf models into our pipelines as preprocessors, postprocessors, or evaluation components
Build model-serving endpoints (e.g., with Gradio or FastAPI) to enable researchers and internal users to experiment with models interactively
Design, implement, and maintain evaluation pipelines, providing high-quality tooling to the broader team to measure model quality and track improvements
Improve configuration hygiene and reproducibility using systems like Hydra, and ensure smooth overrides, templates, and environment switching
Lead efforts in packaging and release of Python modules using modern tools (uv, just, pydantic) for both OSS and internal consumption
Set the standard for code health, test coverage, and release readiness across the team. Write documentation and automation to scale good practices
Qualification
Required
Expert-level proficiency in Python, with a strong foundation in modular design, abstraction boundaries, and collaborative codebase evolution
Fluency with PyTorch, including the ability to run, debug, and patch inference-time model behavior in research-level codebases. Comfort modifying pre/post-processors, model wrappers, and checkpoint logic
Proven experience in refactoring large codebases—cleaning up legacy implementations, eliminating anti-patterns, and paying down tech debt to improve long-term maintainability
Strong grasp of configuration systems, especially Hydra, with an emphasis on reproducibility, override logic, and environment scoping
Familiarity with Python packaging tools like uv, just, and pydantic, including experience managing environment consistency and shipping libraries as artifacts
Strong instincts around code health: API design, directory structure, writing unit and integration tests, exception hygiene, docstrings, and dependency isolation
Comfortable deploying models internally via Gradio or similar frameworks to enable interactive evaluation and feedback from researchers or downstream users
BS or MS (or equivalent experience) in Computer Science, Software Engineering, or a related technical field and 10+ years of industry experience
Preferred
Proficiency in model configs, especially Hydra! Comfortable crafting hierarchical config systems with reusable templates, environment scoping, and overrides for evaluation, inference, or release
Prior work cleaning up sophisticated generative model codebases—adding tests, improving wrappers, and instrumenting code for observability and debugging
Demonstrated success raising engineering quality in a research setting: taking exploratory code and evolving it into a robust, production-friendly module
Track record of mentoring teammates on software engineering best practices and proactively identifying long-term structural risks in fast-moving teams
Passion for building ML tooling that is not only functional, but also elegant, intuitive, and maintainable by others
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
Deccan Chronicle
2025-12-31
IEEE Spectrum
2025-12-31
Company data provided by crunchbase