• Distinguished Engineer , JAX

    NVIDIA (Santa Clara, CA)
    NVIDIA is hiring a distinguished engineer to develop its AI platform and more specifically its performance optimizations in deep learning frameworks using JAX ... for a wide range of deep learning solutions using JAX . To achieve this and provide an excellent user...Play significant role in NVIDIA's effort in contributing to JAX . + Design and implement JAX core… more
    NVIDIA (01/13/26)
    - Related Jobs