Facebook pixel

Senior Software Engineer, Jax
Posted on 4/7/2022
INACTIVE
Locations
San Jose, CA, USA • Remote
Experience Level
Entry
Junior
Mid
Senior
Expert
Desired Skills
Pytorch
Research
Tensorflow
Python
Scikit-Learn
Requirements
  • BS in Computer Science or Computer Engineering or related field (or equivalent experience)
  • C/C++ and Python programming
  • Experience with machine learning frameworks and their internals (e.g. PyTorch, TensorFlow, scikit-learn, etc.)
  • 6+ years of experience developing customer-facing solutions, balancing feature requests and bugs
  • Proven technical foundation in CPU and GPU architectures, numeric libraries, modular software design
  • Highly motivated with excellent verbal and written communication skills
  • Ability to work successfully with multi-functional teams, principles and architects. Coordinates effectively across organizational boundaries and geographies
Responsibilities
  • Play a significant role in NVIDIA's effort in to contributing to JAX
  • Design and implement core components of PyTorch and TensorFlow to effective use JAX and drive peak GPU performance
  • 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 deep learning research and its application in real world products
Desired Qualifications
  • MS or PhD in Computer Science or Computer Engineering or related field
  • Understanding of JAX and Autograd
  • Understanding of deep learning training in distributed contexts: multi-GPU, multi-node, synchronous vs asynchronous
  • Background with software shipping cycles (dev, deploy, release, CI)
  • Experience building distributed systems and services at large scale
NVIDIA

10,001+ employees

Designer & manufacturer of computer chips & graphics processors
Company mission
NVIDIA is on a mission to solve the world's most stimulating technology problems – in industries ranging from gaming to scientific exploration.
Company Values
  • Innovation
  • Speed & Agility
  • Intellectual Honesty
  • Excellence
  • One Team