onsite
Senior Staff Software Engineer - AI Applications and Platform Foundations - NVIDIA
Software Engineer
Lead the design and delivery of production‑grade AI applications and shared platforms, leveraging GPU, CUDA, and distributed systems to power enterprise AI across desktop, mobile, web, and cloud.
About the role
Key Responsibilities
- Architect and implement scalable AI application frameworks that run on GPU‑accelerated hardware across multiple platforms.
- Drive performance optimization of deep learning pipelines using CUDA, C++, and Python, ensuring sub‑second inference and high throughput training.
- Collaborate with cross‑functional teams to integrate AI services into enterprise products, from data ingestion to model deployment.
- Mentor and lead a team of software engineers, setting coding standards, conducting code reviews, and fostering a culture of continuous improvement.
- Research and prototype emerging AI techniques, translating research into production‑ready solutions.
Requirements
- 10+ years of software engineering experience with a strong focus on AI and GPU computing.
- Expertise in Python, C++, and CUDA for high‑performance computing.
- Proven track record delivering large‑scale, distributed AI systems in production.
- Deep understanding of machine learning frameworks (e.g., TensorFlow, PyTorch) and model optimization.
- Excellent communication skills and ability to influence stakeholders across technical and non‑technical domains.
Skills
pythonccudamachine learningdeep learning