Senior Generative-AI Software Engineer

We are the Picasso team and we are seeking a Senior Engineer to join our rapidly growing team and contribute to our Generative AI solutions. NVIDIA has been continually revolutionizing accelerated computing and powering the engines of artificial intelligence. We are developing a range of solutions, including text-to-image, text-to-video, and text-to-3D, that are trained using clean data provided by our industry partners and deployed for quick inference. Come join us in these thrilling times and make a significant impact in the burgeoning field of Machine Learning! Achieving what's never been done before requires vision, innovation, and top-notch talent. As an NVIDIAN, you'll be part of a diverse, nurturing environment where everyone is inspired to do their life's work. What you will be doing: - Study and work on innovative image-based Generative AI models. Ensure the best possible performance on current and future GPU architectures. - Architect, design, and utilize your software engineering skills to optimize, deploy, and commercialize AI models. Develop clean, efficient, well-tested production code. - Collaborate closely with Research, Cloud, Architecture, and Libraries teams at NVIDIA to implement inferencing and fine-tuning solutions. What we need to see: - Understanding of contemporary concepts in Deep Neural Networks, Generative AI, and Diffusion Models in particular, ideally backed by significant contributions to open-source projects. - Working knowledge of Docker Containers, REST API, cloud-based Compute and Storage. - A Bachelor's or Master's Degree in Engineering or Computer Science or a related field or equivalent experience. - 8+ years of relevant work experience. 5+ years in Machine Learning frameworks such as PyTorch, JAX, or Tensorflow, ONNX Runtime, and TensorRT. - Solid understanding of C++, Python, and parallel programming. - Exceptional communication and interpersonal skills, along with the ability to work in a fast-paced, distributed team. Prior experience mentoring junior engineers and interns is a plus. - Strong computer science fundamentals - algorithms and data structures, computational complexity, parallel and distributed computing, system software. The base salary range is 176,000 USD - 333,500 USD. Your base salary will be determined based on your location, experience, and the wages of employees in similar positions. You will also be eligible for equity and benefits.