Senior Software Engineer, JAX
USonsitesenior
Posted today · via Workday
About this role
NVIDIA is hiring senior engineers to develop its AI platform and more specifically its performance optimizations in deep learning frameworks using JAX, a tool that can differentiate between Python control code and NumPy code. Our goal is to deliver a polished, fast, modular and coordinated platform to handle data, training and analysis for a wide range of deep learning solutions using JAX. To achieve this and provide an excellent user experience, you have strong programming skills and excellent system design skills. You should have excellent communication and planning skills for this role.…
What we'd score you on
reqspace match rubricFive dimensions, recruiter-grade. Upload your resume and we'll generate a written explanation of where you fit and where the gaps are.
1
Skills match
For this role: python, c++, numpy, scikit-learn, pytorch…
2
Level fit
This role is senior-level. We check your trajectory against it.
3
Domain experience
Your work in the role's domain matters more than your years total. We weight recent and direct experience.
4
Recency
A skill you used last quarter weighs more than one from five years ago. We grade on recency, not lifetime.
5
Location fit
This role is based in US. We weight your proximity and willingness to relocate.
Score yourself on this role.
Free · no card · written explanation included
Skills in this role
Pulled from the job description. These are the keywords we'll weight when scoring your fit.
pythonc++numpyscikit-learnpytorchtensorflowjaxteams
More at Nvidia
- View →Senior System Software EngineerIndia, Pune
- View →Senior Deep Learning Researcher, LLM InferenceIsrael, Tel Aviv
- View →Dell Technologies Global Account ManagerUS, CA, Santa Clara
- View →Director, Business Development - Retail Intelligent StoresUS, CA, Santa Clara
- View →Manager, Silicon Operations Production PlanningUS, CA, Santa Clara
- View →Go-to-Market Product ManagerUS, CA, Santa Clara