JAX Performance Architect for AI Platform

A leading technology company in Santa Clara is seeking a Distinguished Engineer to develop AI platform tools and optimize performance in deep learning frameworks using JAX. Candidates must have extensive programming experience in C/C++ and Python, alongside 18+ years in related fields. The role involves designing JAX core components, collaborating with researchers, and ensuring the delivery of high-quality AI solutions. This position offers a competitive salary and is committed to fostering a diverse work environment. #J-18808-Ljbffr