Distinguished Engineer, JAX

Job Summary

A company is looking for a Distinguished Engineer, JAX.

Key Responsibilities

• Contribute significantly to the development of JAX and its core components
• Design and implement performance optimizations for NVIDIA products
• Collaborate with AI researchers to build efficient AI-based systems and tools

Required Qualifications

• BS in Computer Science, Computer Engineering, or related field (or equivalent experience)
• 18+ years of relevant experience
• Proficiency in C/C++ and Python programming
• Experience with machine learning frameworks and their internals
• Strong technical foundation in CPU/GPU architectures and modular software design