NVIDIA is hiring senior engineers to develop its AI platform using JAX and improve performance optimizations in deep learning frameworks.
Requirements
- BS in Computer Science or Computer Engineering or related field (or equivalent experience)
- 6+ years relevant experience
- C/C++ and Python programming
- Experience with machine learning frameworks and their internals (e.g. PyTorch, TensorFlow, scikit-learn, etc.)
- Proven ability 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
Benefits
- Base salary determined by location, experience, and pay of employees in similar positions
- Determined by location, experience, and pay of employees in similar positions