At NVIDIA, we're not just building the future, we're generating it! Our Cosmos generative AI engineering team is pushing the boundaries of what’s possible across multimodal learning, video generation, synthetic data, intelligent simulation, and agentic systems. We are looking for exceptionally driven engineers and applied scientists with deep experience in generative modeling to help define the next era of AI computing. What you'll be doing: 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.