Senior Software Engineer, JAX

Found in: beBee S US - 1 week ago

Santa Clara, United States NVIDIA Full time
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. Come join us to build the next generation of tooling that will advance the numerical computing and machine learning research around the world

What You Will Be Doing:

Play meaningful role in NVIDIA's effort in contributing to JAX.

Design and implement JAX core components and drive peak performance on NVIDIA products.

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 numerical computing, simulation and deep learning research and their applications in real world products.

What We Need To See:

BS in Computer Science or Computer Engineering or related field (or equivalent experience)

5+ 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. Coordinates effectively across organizational boundaries and geographies.

Ways To Stand Out From The Crowd:

Understanding of JAX, Autograd, tracing, code generation and DSL compilers and their design.

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.

The base salary range is 180,000 USD - 339,250 USD. Your base salary will be determined based on your location, experience, and the pay of employees in similar positions.

You will also be eligible for equity and benefits. NVIDIA accepts applications on an ongoing basis.

NVIDIA is committed to fostering a diverse work environment and proud to be an equal opportunity employer. As we highly value diversity in our current and future employees, we do not discriminate (including in our hiring and promotion practices) on the basis of race, religion, color, national origin, gender, gender expression, sexual orientation, age, marital status, veteran status, disability status or any other characteristic protected by law.

  • Senior Software Engineer, JAX

    1 week ago

    Santa Clara, United States NVIDIA Full time

    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...

  • Senior Software Engineer, JAX

    2 weeks ago

    Santa Clara, United States NVIDIA Full time

    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...

  • Senior Software Engineer, JAX

    Found in: Jooble US O C2 - 1 day ago

    Santa Clara, CA, United States NVIDIA Full time

    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...

  • Senior Software Engineer, JAX

    2 weeks ago

    Santa Clara, CA, United States NVIDIA Full time

    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...

  • Senior Software Engineer, Data Platform

    Found in: Jooble US O C2 - 1 day ago

    Santa Clara, CA, United States NVIDIA Full time

    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...

  • Senior Software Engineer

    20 hours ago

    Santa Clara, United States Tech Firefly Full time

    We are offering excellent opportunities for Senior Software EngineersApplicants are required to be eligible to lawfully work in the U.S. immediately; employer will not transfer or sponsor applicants for U.S. work authorization (such as an H-1B visa) for this opportunity.Direct Hires Only.No Recruiters or Solicitations.Job Type: Full-timeWork...

  • Senior Software Engineer

    1 week ago

    Santa Clara, United States Tech Firefly Full time

    Job DescriptionJob DescriptionWe are offering excellent opportunities for Senior Software EngineersApplicants are required to be eligible to lawfully work in the U.S. immediately; employer will not transfer or sponsor applicants for U.S. work authorization (such as an H-1B visa) for this opportunity.Direct Hires Only.No Recruiters or Solicitations.Job Type:...

  • Senior Software Engineer

    Found in: Appcast Linkedin GBL C2 - 2 weeks ago

    Santa Clara, United States Tech Firefly Full time

    We are offering excellent opportunities for Senior Software EngineersApplicants are required to be eligible to lawfully work in the U.S. immediately; employer will not transfer or sponsor applicants for U.S. work authorization (such as an H-1B visa) for this opportunity.Direct Hires Only.No Recruiters or Solicitations.Job Type: Full-timeWork...

  • Software Engineer in Test

    2 weeks ago

    Santa Clara, United States Intelliswift Software Full time

    Job Title: Software Development Engineer in Test (SDET)Client Location: Santa Clara, CA 95054 (50% Hybrid schedule)Duration: 08 months contract with possible extensionJob Description:Software Development Engineers in Test are involved in all aspects of quality assurance activities for a scrum team. Working closely with other Quality Engineers, SDETs,...

  • Software Engineer in Test

    Found in: Appcast Linkedin GBL C2 - 2 weeks ago

    Santa Clara, United States Intelliswift Software Full time

    Job Title: Software Development Engineer in Test (SDET)Client Location: Santa Clara, CA 95054 (50% Hybrid schedule)Duration: 08 months contract with possible extensionJob Description:Software Development Engineers in Test are involved in all aspects of quality assurance activities for a scrum team. Working closely with other Quality Engineers, SDETs,...

  • Senior Software Architecture Engineer

    5 days ago

    Santa Clara, United States ATR International Full time

    The rapid acceleration of deep learning and generative AI is leading to new generations of high-performance and low-power chip architecture. Our architecture engineers work at the forefront of this technological leap in computer architecture. Our team members translate algorithms into hardware design specifications, identify performance bottlenecks, optimize...

  • Senior Software Architecture Engineer

    Found in: Appcast US C2 - 1 week ago

    Santa Clara, United States ATR International Full time

    The rapid acceleration of deep learning and generative AI is leading to new generations of high-performance and low-power chip architecture. Our architecture engineers work at the forefront of this technological leap in computer architecture. Our team members translate algorithms into hardware design specifications, identify performance bottlenecks, optimize...

  • Senior Software Architecture Engineer

    Found in: Appcast Linkedin GBL C2 - 1 week ago

    Santa Clara, United States ATR International Full time

    The rapid acceleration of deep learning and generative AI is leading to new generations of high-performance and low-power chip architecture. Our architecture engineers work at the forefront of this technological leap in computer architecture. Our team members translate algorithms into hardware design specifications, identify performance bottlenecks, optimize...

  • Software Engineer in Test

    6 days ago

    Santa Clara, United States Intelliswift Software Inc Full time

    Job Title: Software Development Engineer in Test (SDET) Client Location: Santa Clara, CA 95054 (50% Hybrid schedule) Duration: 08 months contract with possible extension Job Description: Software Development Engineers in Test are involved in all aspects of quality assurance activities for a scrum team. Working closely with other Quality Engineers, SDETs,...

  • Senior Software Engineer

    Found in: Resume Library US A2 - 1 week ago

    Santa Clara, California, United States Tech Firefly Full time

    We are offering excellent opportunities for Senior Software Engineers Applicants are required to be eligible to lawfully work in the U.S. immediately; employer will not transfer or sponsor applicants for U.S. work authorization (such as an H-1B visa) for this opportunity. Direct Hires Only. No Recruiters or Solicitations. Job Type: Full-time Work...

  • Senior Software Engineer

    1 week ago

    Santa Clara, United States Tech Firefly Full time

    Job DescriptionJob DescriptionSummary:The Project Manager is an Information Technology (IT) position and will direct, oversee, monitor, and report to our client. The Project Manager will collaborate with project stakeholders and implement the project work plan to achieve project objectives. The Project Manager will lead daily Agile Scrum standup meetings and...

  • Senior Software Engineer

    Found in: Appcast Linkedin GBL C2 - 2 weeks ago

    Santa Clara, United States Tech Firefly Full time

    Summary:The Project Manager is an Information Technology (IT) position and will direct, oversee, monitor, and report to our client. The Project Manager will collaborate with project stakeholders and implement the project work plan to achieve project objectives. The Project Manager will lead daily Agile Scrum standup meetings and help identify and resolve...

  • Senior Graphics Software Engineer

    4 days ago

    Santa Clara, United States CareerBuilder Full time

    We are looking for a Senior Graphics Software Engineer. Nvidias invention of the GPU 1999 sparked the growth of the PC gaming market, redefined modern computer graphics, and revolutionized parallel computing. More recently, GPU deep learning ignited modern AI - the next era of computing - with the GPU acting as the brain of computers, robots, and...

  • Senior Graphics Software Engineer

    5 days ago

    Santa Clara, United States NVIDIA Full time

    NVIDIA's invention of the GPU in 1999 fueled the growth of the PC gaming market, redefined modern computer graphics, and revolutionized parallel computing. More recently, GPU deep learning ignited modern AI — the next era of computing — with the GPU acting as the brain of computers, robots, and self-driving cars that can perceive and understand the...

  • Senior Software Engineer

    Found in: Resume Library US A2 - 1 week ago

    Santa Clara, California, United States ServiceNow Full time

    Job Description Come join the Cloud Automation Platform team and work with a talented group of developers building out solutions that make our infrastructure highly available, scalable and fully automated. As a Senior Software Engineer focused on cloud automation, you will have the opportunity to work on ServiceNow Platform and Tools, build solutions using...