Senior Software Engineer, JAX

Senior software engineer position to develop AI platform and performance optimizations in deep learning frameworks using JAX. Polished, fast, modular, and coordinated platform for handling data, training, and analysis for deep learning solutions. Excellent programming and system design skills required.

Requirements

• Play meaningful role in NVIDIA's effort in contributing to JAX
• Design and implement JAX core components and drive peak performance on NVIDIA products
• Work with AI applied researchers and leaders to build future-proof models
• Build tools that will increase the efficiency of teams developing AI-based systems
• Work to bridge the gap between the latest in numerical computing, simulation, and deep learning research and their applications in real-world products

Benefits

• Equity
• Benefits