Current jobs related to Senior Software Engineer, JAX - Santa Clara - Nvidia

  • Senior Research Software Engineer

    5 days ago

    Santa Clara, California, United States ServiceNow Full time

    Job Description**About ServiceNow**ServiceNow is a global market leader in the field of cloud-based platforms, bringing innovative AI-enhanced technology to over 8,100 customers, including 85% of the Fortune 500. Our intelligent cloud-based platform seamlessly connects people, systems, and processes to empower organizations to find smarter, faster, and...

  • Senior Software Architect for AI Resilience

    7 days ago

    Santa Clara, California, United States NVIDIA Full time

    About the RoleNVIDIA is seeking a highly skilled Senior Software Architect to lead the development of AI software resilience for our most powerful AI supercomputers.Key ResponsibilitiesDevelop and implement critical resilience features to support frontier model training at scale, ensuring robust and reliable AI systems.Serve as a trusted authority on AI...

  • Senior Performance Engineer

    5 days ago

    Santa Clara, United States NVIDIA Full time

    Senior Performance Engineer - Deep Learning NVIDIA is hiring senior software engineers to build and optimize the tools Deep Learning engineers use across the world to design, develop, and deploy AI applications. We are an ambitious, forward-thinking and diverse team that influences all areas of NVIDIA's AI platform and directly contributes to premiere Deep...

  • Senior AI Resiliency Software Engineer

    5 days ago

    Santa Clara, California, United States NVIDIA Full time

    Job DescriptionWe are seeking a highly skilled Principal Engineer for AI Software Resiliency to join our team at NVIDIA.Key ResponsibilitiesLead the development of AI software resiliency features for our most powerful AI supercomputers.Collaborate with multiple teams and stakeholders to align on mission requirements and ensure successful integration of...

  • Principal Software Engineer for AI Resiliency

    5 days ago

    Santa Clara, California, United States NVIDIA Full time

    Job DescriptionWe are seeking a highly skilled Principal Engineer for AI Software Resiliency to join our team at NVIDIA. As a key member of our organization, you will play a pivotal role in defining and implementing critical resiliency features for AI supercomputers at a scale of 100,000+ GPUs.Key ResponsibilitiesDevelop and lead the execution of software...

  • Principal Engineer for AI Software Resiliency

    1 week ago

    Santa Clara, United States NVIDIA Full time

    Principal Engineer for AI Software Resiliency We are now looking for a Principal Engineer for AI Resiliency. At NVIDIA, we are pushing the boundaries of what’s possible in AI. We are seeking a Principal Software Engineer to lead the development of AI software resiliency for the most powerful AI supercomputers in the world. As a lead focused on AI Software...

  • Lead Software Engineer for AI Supercomputing

    5 days ago

    Santa Clara, California, United States NVIDIA Full time

    We are seeking a highly skilled Principal Software Engineer to lead the development of AI software resiliency for the most powerful AI supercomputers in the world.As a lead focused on AI Software Resiliency, you will play a pivotal role in defining and implementing critical resiliency features for AI supercomputers at a scale of 100,000+ GPUs.Your expertise...

  • Senior Scientific Machine Learning Software Engineer

    1 month ago

    Santa Clara, California, United States Nvidia Full time

    Senior Scientific Machine Learning Software Engineer - PhysicslocationsUS, CA, Santa ClaraUS, Remotetime typeFull timejob requisition idJR1981550NVIDIA's deep learning and HPC platforms have made a huge impact in various fields and are broadly used across leading academic institutions, start-ups, and industry, including the world's largest Internet...

  • Senior Scientific Machine Learning Software Engineer

    2 months ago

    Santa Clara, United States Nvidia Full time

    Senior Scientific Machine Learning Software Engineer - PhysicslocationsUS, CA, Santa ClaraUS, Remotetime typeFull timejob requisition idJR1981550NVIDIA’s deep learning and HPC platforms have made a huge impact in various fields and are broadly used across leading academic institutions, start-ups, and industry, including the world’s largest Internet...

  • Senior Software Test Engineer

    6 hours ago

    Santa Clara, United States Denken Solutions, Inc. Full time

    Title: Senior Software Test Engineer - (L2 & L3 Protocols)Full-Time role (Open for Relocation)Location: Santa Clara, CAJob duties: Design, develop and execute test plans for testing software features on Layer 3 protocols such as IPV4, IPV6, OSPF, BGP, L3VPN and Multicast protocols such as IGMP and PIM. Design, develop and execute test plans for testing Layer...

  • Senior Software Engineer

    2 weeks ago

    Santa Clara, United States AIxMed, Inc. Full time

    Job Title: Senior Software Engineers – AI/Computer Vision: Computational CytopathologySummary:Embark on a journey at the cutting edge of digital cytopathology with AIxMed, a pioneering startup leveraging AI and computer vision to transform digital cytology. In the role of Senior Software Engineer, you will be instrumental in crafting innovative software...

  • Senior Software Engineer

    3 weeks ago

    Santa Clara, United States AIxMed, Inc. Full time

    Job Title: Senior Software Engineers – AI/Computer Vision: Computational CytopathologySummary:Embark on a journey at the cutting edge of digital cytopathology with AIxMed, a pioneering startup leveraging AI and computer vision to transform digital cytology. In the role of Senior Software Engineer, you will be instrumental in crafting innovative software...

  • Senior Principal Software Engineer

    2 days ago

    Santa Clara, California, United States Palo Alto Networks Full time

    About the RoleWe are seeking a highly skilled and experienced Senior Principal Software Engineer to join our team at Palo Alto Networks. As a key member of our engineering team, you will be responsible for designing, building, and optimizing our software products to meet the needs of our customers.Key ResponsibilitiesLead the development of complex software...

  • Senior Software Engineer

    4 weeks 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

    3 months 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

    3 weeks ago

    Santa Clara, United States ALOIS Solutions Full time

    Job Title: Senior Software Engineer/ Embedded Engineer/ Member of Technical StaffLocation: Santa Clara, California, 95054 (4 days onsite and 1 day remote)Pay range: $60 to $80 per hour on W2 onlyThis is a contract position on W2 only, no C2C or H1b Summary: • Full life cycle software development from design through development, validation and release.•...

  • Senior Software Engineer

    3 weeks ago

    Santa Clara, United States ALOIS Solutions Full time

    Job Title: Senior Software Engineer/ Embedded Engineer/ Member of Technical StaffLocation: Santa Clara, California, 95054 (4 days onsite and 1 day remote)Pay range: $60 to $80 per hour on W2 onlyThis is a contract position on W2 only, no C2C or H1b Summary: • Full life cycle software development from design through development, validation and release.•...

  • Senior Imaging Systems Software Engineer

    3 months ago

    Santa Clara, United States DeepSight Technology Full time

    Job DescriptionJob DescriptionSenior Imaging Systems Software EngineerDeepSight Technology Inc. is actively searching for a Senior Imaging Systems Software Engineer to apply to join our amazing software team full-time in our Santa Clara, CA office!We are an innovative medical imaging company that invests in our team and offers real opportunities for career...

  • Senior Software Engineer

    2 weeks ago

    Santa Clara, California, United States Oracle Full time

    Job OverviewAs a Senior Software Engineer at Oracle, you will collaborate with teams of developers tasked with the design, creation, and maintenance of our innovative software solutions. Your role will require you to architect comprehensive system interactions, engage in hands-on development, and possess a deep understanding of cloud infrastructure and...

  • Senior Software Engineer, 3D

    1 month ago

    Santa Clara, United States UnitX Full time

    Job Title:Senior Software Engineer, 3DAbout Us:UnitX builds the world's leading robotics systems to automate repetitive visual tasks in factories. UnitX is an exciting, innovative, and fast-moving startup with a team from Stanford, MIT, and Google. Our team is highly collaborative and we are looking forward to offering you the chance to manifest your...

Senior Software Engineer, JAX

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