Senior Software Engineer, Data Platform

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



  • Santa Clara, United States SoundHound Full time

    SOUNDHOUND INC. TURNS SOUND INTO UNDERSTANDING AND ACTIONABLE MEANING. We believe in enabling humans to interact with the things around them in the same way we interact with each other: by speaking naturally to mobile phones, cars, TVs, music speakers, coffee machines, and every other part of the emerging 'connected' world. Our latest product, Hound,...


  • Santa Clara, CA, United States Nvidia Full time

    NVIDIA is looking for Senior Software Engineering to join NVIDIA in the Cumulus Linux team! We are the team that develops the Network Operating System that powers data centers that are accelerated, disaggregated and software-defined. We are delivering this high demand product to lead the exploding growth in AI and high-performance computing. You will play a...

  • Sr Software Engineer

    3 weeks ago


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

    Job Description Your Career  Palo Alto Networks® SaaS Security team is looking for a seasoned and accomplished Senior Staff Software Engineer to help scale out our security platform with a sharp focus on platform and infrastructure capabilities.  As a member of the team, you have the unique opportunity to:   Be part of a world-class software...


  • Santa Clara, United States MapR Technologies Full time

    ABOUT MAPR TECHNOLOGIES MapR Technologies, a provider of the industry’s leading data platform for AI and Analytics, enables enterprises to inject analytics into their business processes to increase revenue, reduce costs, and mitigate risks. MapR addresses the data complexities of high-scale and mission-critical distributed processing from the cloud to the...


  • Santa Clara, United States NVIDIA Full time

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


  • Santa Clara, United States NVIDIA Full time

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


  • Santa Clara, United States NVIDIA Full time

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


  • Santa Clara, United States Ventrum Full time

    Looking for: Senior Manager, Data PlatformJob Type: Full timeLocation: Santa Clara, CA (Hybrid)Position Overview: The Senior Manager, Data Platform will play a pivotal role in driving data-based decision-making culture across the company. You will be responsible for overseeing the development and management of our data platform strategy and implementation,...


  • Santa Clara, United States Ventrum Full time

    Looking for: Senior Manager, Data PlatformJob Type: Full timeLocation: Santa Clara, CA (Hybrid)Position Overview: The Senior Manager, Data Platform will play a pivotal role in driving data-based decision-making culture across the company. You will be responsible for overseeing the development and management of our data platform strategy and implementation,...


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


  • San Francisco, CA, United States Discord Full time

    Discord is about giving people the power to create space to find belonging in their lives. We want to make it easier for you to talk regularly with the people you care about. We want you to build genuine relationships with your friends and communities close to home or around the world. Original, reliable, playful, and relatable. These are the values that...


  • Santa Clara, United States Protingent Full time

    WHAT WE NEED FROM YOU AS A SENIOR IMAGING SYSTEMS SOFTWARE ENGINEERWe are looking for an engineer with a strong background in ultrasound, signal processing, image optimization, and programming languages such as Python or C++ to shape our new ultrasound imaging platform. As a Senior Imaging Systems Software Engineer, you will work on characterization,...


  • Santa Clara, United States Protingent Full time

    WHAT WE NEED FROM YOU AS A SENIOR IMAGING SYSTEMS SOFTWARE ENGINEERWe are looking for an engineer with a strong background in ultrasound, signal processing, image optimization, and programming languages such as Python or C++ to shape our new ultrasound imaging platform. As a Senior Imaging Systems Software Engineer, you will work on characterization,...


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

    Job Description Your Career As a member of the Platform Software group at Palo Alto Networks, you will participate in the design and development of platform software of our Next-Generation Firewall platforms. Responsibilities will include platform bring-up  (both pizza-box and chassis-based platforms), platform security using TPM chip, development of both...


  • Santa Clara, United States Petlibro Full time

    About the jobAbout PetlibroPetlibro is a design thinking company creating products that nurture the intertwined lives of pets & their people. We launched with a philosophy that good design, in form & in function, can make a difference. Petlibro innovates with the latest technology to solve everyday problems for the modern pet parents and revolutionize how we...


  • Santa Clara, United States Petlibro Full time

    About the jobAbout PetlibroPetlibro is a design thinking company creating products that nurture the intertwined lives of pets & their people. We launched with a philosophy that good design, in form & in function, can make a difference. Petlibro innovates with the latest technology to solve everyday problems for the modern pet parents and revolutionize how we...


  • San Mateo, CA, United States Snowflake Computing Full time

    Build the future of data. Join the Snowflake team.Are you interested in building the only one Data Cloud that unlocks near-unlimited data opportunities? Are you excited about building systems with near-infinite scalability, performance and flexibility? Would you like to be part of one of the fastest-growing companies with tremendous growth opportunities?...


  • Santa Clara, United States ATR International Full time

    We are seeking a Senior Software Architecture Engineer for a very important client Job Description The rapid acceleration of deep learning and generative AI is leading to new generations of high-performance and low-power chip architecture At the Company, architecture engineers work at the forefront of this technological leap in computer architecture Our...


  • Redwood City, CA, United States C3 IoT Full time

    C3.ai, Inc. (NYSE:AI) is a leading provider of Enterprise AI software for accelerating digital transformation. The proven C3 AI Platform provides comprehensive services to build enterprise-scale AI applications more efficiently and cost-effectively than alternative approaches. The core of the C3 AI offering is an open, data-driven AI architecture that...


  • San Mateo, CA, United States Snowflake Computing Full time

    Build the future of data. Join the Snowflake team.The growth platform team’s charter is to grow our on demand business and make everyone big and small successful on Snowflake,  help our customers utilize the power of Snowflake for their businesses. The team works on enhancing the foundation, infrastructure and the full journey of user experiences to help...