Software Engineer, Machine Learning Inference - New… at NVIDIA · US

Summary: NVIDIA is at the forefront of innovation, driving advancements in AI and machine learning. They are seeking talented engineers to join their TensorRT team to develop deep learning inference software for NVIDIA AI accelerators. Responsibilities: - Design, develop and optimize NVIDIA TensorRT and TensorRT-LLM to supercharge inference applications for datacenter, workstations, and PCs - Develop software in C++, Python, and CUDA for seamless and efficient deployment of state-of-the-art LLMs and Generative AI models - Collaborate with deep learning experts and GPU architects throughout the company to influence Hardware and Software design for inference Required Qualifications: - BS, MS, PhD or equivalent experience in Computer Science, Computer Engineering or a related field - Software development experience on a large codebase or project - Strong proficiency in C++ (required), Rust or Python programming languages - Experience in developing Deep Learning Frameworks, Compilers, or System Software - Excellent problem-solving skills and passion to learn and work effectively in a fast-paced, collaborative environment - Strong communication skills and the ability to articulate complex technical concepts Preferred Qualifications: - Experience in developing inference backends and compilers for GPUs - Knowledge of Machine Learning techniques and GPU programming with CUDA or OpenCL - Background in working with LLM inference frameworks like TensorRT-LLM, vLLM, SGLang - Experience working with deep learning frameworks like TensorRT, PyTorch, JAX - Knowledge of close-to-metal performance analysis, optimization techniques, and tools Required Skills: C++, Python, Rust, CUDA, Deep Learning Frameworks, Compilers, System Software, Machine Learning, GPU programming, TensorRT, TensorRT-LLM, PyTorch, JAX Benefits: Equity, Benefits