NVIDIA is a world-leader in artificial intelligence and computer vision. Our team builds hardware-accelerated computer vision pipelines, cloud services and SDKs bringing the latest AI innovations to data centers, gaming rigs, cars, robots, buildings, medical devices, and more. We are looking for an expert in system-level software optimization to push our computer vision applications to near speed of light. The right candidate will bring insight into challenges of delivering performance at scale and passion for multi-disciplinary teamwork and efficient, well-crafted software.
What you'll be doing:
- Develop, profile and optimize data-center and edge computer vision workloads for efficiency, latency, and throughput (Python).
- Implement and improve computer vision and image processing algorithms using CUDA.
- Upstream performance improvements to SDKs and libraries across NVIDIA to deliver accelerated computer vision at scale.
- Influence software architecture, validation strategy and technical roadmaps to ensure outstanding performance.
- Promote high-performance computer vision across NVIDIA teams and functions (Engineering, Product Management, Marketing, and more).
What we need to see:
- Master's of Science in Computer Science or Electrical engineering or equivalent experience.
- 8 years of practical experience.
- Excellent software engineering fundamentals (source control, CI/CD, testing/validation, packaging, containerization, release). Proven track record developing, testing and releasing production-grade, complex software.
- Proficiency with Python, CUDA and C++.
- Strong fundamentals with multi-threaded, multi-process and distributed software development. Expertise defining and driving performance metrics through profiling and benchmarking.
- Experience developing performance-critical data center and cloud applications (REST APIs, gRPC).
Ways to Stand Out from the Crowd:
- Expertise in classical, non-ML computer vision
- Expertise in ML computer vision (VLMs, Vision Transformers, Diffusion models) and its software ecosystem: PyTorch, HuggingFace, vLLM
- Grounding in mathematical fundamentals such as linear algebra, numerical methods, statistics, and exploratory data analysis.
- History of creativity and innovation around performance in multiple problem domains.
#LI-Hybrid
Your base salary will be determined based on your location, experience, and the pay of employees in similar positions. The base salary range is 184,000 USD - 287,500 USD for Level 4, and 224,000 USD - 356,500 USD for Level 5.
You will also be eligible for equity and benefits.
Applications for this job will be accepted at least until September 7, 2025.NVIDIA is committed to fostering a diverse work environment and proud to be an equal opportunity employer. As we highly value diversity in our current and future employees, we do not discriminate (including in our hiring and promotion practices) on the basis of race, religion, color, national origin, gender, gender expression, sexual orientation, age, marital status, veteran status, disability status or any other characteristic protected by law.
