Staff Software Engineer, Deep Learning Acceleration

Job Description

The role involves conducting performance analysis and optimization of deep learning networks for autonomous vehicle systems. You will optimize software architecture and latency for both onboard vehicle deployment and large-scale data center training.

Requirements: Candidates need at least 5 years of software engineering experience and a degree in Computer Science or a related field. Proficiency in CUDA, C++, Python, and deep learning frameworks like PyTorch or TensorFlow is required.

Key Skills: CUDA, C++, Python, High-performance computing, Parallel programming, NVIDIA Nsight Systems, Nsight Compute, Roofline model, PyTorch, TensorFlow, Computer vision, Transformer-based architectures, Linux/Unix, TensorRT, OpenAI Triton, Mojo

Benefits: Annual bonus, Equity compensation, Benefits