Senior Deep Learning Software Engineer

4 weeks ago


Los Angeles, California, United States Nvidia Full time
Job Title: Senior Deep Learning Software Engineer

We are seeking a highly skilled Senior Deep Learning Software Engineer to join our team at NVIDIA. As a key member of our model optimization group, you will play a pivotal role in designing and building our automated inference and deployment solution.

Key Responsibilities:
  • Architect and design a modular and scalable software platform to provide an excellent user experience with broad model support and optimization techniques.
  • Leverage and build upon the torch 2.0 ecosystem to analyze and extract standardized model graph representation from arbitrary torch models for our automated deployment solution.
  • Develop high-performance optimization techniques for inference, such as automated model sharding techniques, efficient attention kernels with kv-caching, and more.
  • Collaborate with teams across NVIDIA to use performant kernel implementations within the automated deployment solution.
  • Analyze and profile GPU kernel-level performance to identify hardware and software optimization opportunities.
Requirements:
  • Masters, PhD, or equivalent experience in Computer Science, AI, Applied Math, or related field.
  • 5+ years of relevant work or research experience in Deep Learning.
  • Excellent software design skills, including debugging, performance analysis, and test design.
  • Strong proficiency in Python, PyTorch, and related ML tools.
  • Strong algorithms and programming fundamentals.
  • Good written and verbal communication skills and the ability to work independently and collaboratively in a fast-paced environment.
Preferred Qualifications:
  • Contributions to PyTorch, JAX, or other Machine Learning Frameworks.
  • Knowledge of GPU architecture and compilation stack, and capability of understanding and debugging end-to-end performance.
  • Familiarity with NVIDIA's deep learning SDKs such as TensorRT.
  • Prior experience in writing high-performance GPU kernels for machine learning workloads in frameworks such as CUDA, CUTLASS, or Triton.

NVIDIA offers highly competitive salaries and a comprehensive benefits package. If you are a creative, motivated, and love a challenge, we want to hear from you.



  • Los Angeles, California, United States VoxelCloud Full time

    We are seeking a highly skilled Senior Deep Learning Model Developer to join our R&D team at VoxelCloud. As a key member of our team, you will be responsible for developing innovative deep learning solutions for our product lines, including Thorax, Retina, Cardio, and Skin.Key Responsibilities:Design, implement, and test deep learning models for prototyping...


  • Los Angeles, California, United States Google Full time

    About the RoleWe're seeking a highly skilled Senior Software Engineer to join our team at Google. As a key member of our software engineering team, you will be responsible for developing and maintaining complex software systems, with a focus on machine learning and quantum computing.ResponsibilitiesDesign, develop, and test software systems, including...


  • Los Angeles, California, United States Google Full time

    About the RoleWe're seeking a highly skilled Senior Software Engineer to join our team at Google, specializing in machine learning and quantum computing. As a key member of our team, you will be responsible for developing cutting-edge technologies that drive innovation and progress in the field of quantum computing.ResponsibilitiesDesign, develop, and test...


  • Los Angeles, California, United States Kira Learning Full time

    About Kira LearningWe are a leading education technology company creating innovative curriculum and learning tools that empower teachers and prepare students for a world where technology and science play pivotal roles.Our mission is to harness transformative AI technologies to make world-class teaching and learning accessible to everyone.We are seeking a...


  • Los Angeles, California, United States Freeform Full time

    Job Title: Senior Machine Learning EngineerFreeform is revolutionizing the manufacturing industry with its cutting-edge software-defined, autonomous metal 3D printing factories. As a Senior Machine Learning Engineer, you will play a crucial role in designing, developing, and integrating critical data science infrastructure and developing custom machine...


  • Los Angeles, California, United States Freeform Full time

    Job OverviewFreeform is revolutionizing the manufacturing industry with its software-defined, autonomous metal 3D printing factories. As a Senior Machine Learning Engineer, you will play a crucial role in designing, developing, and integrating critical data science infrastructure and developing custom machine learning algorithms from scratch. Your solutions...


  • Los Gatos, California, United States Netflix Full time

    Netflix is a leading entertainment service with 278 million paid memberships in over 190 countries. We're seeking a senior software engineer to join the Metaflow project, developing a product that makes it easy to build, manage, and scale real-life ML applications.As a senior software engineer, you will lead requirements, design, and implementation of...


  • Los Angeles, California, United States Freeform Full time

    Job Title: Senior Machine Learning EngineerFreeform is a pioneering company that is revolutionizing the manufacturing industry with its software-defined, autonomous metal 3D printing factories. Our mission is to make the transformative power of 3D printing available to all industries at scale and unlock the future of innovation.Job SummaryWe are seeking a...


  • Los Angeles, California, United States Medtronic Full time

    About the RoleMedtronic is seeking a highly skilled Senior Software Engineer to join our dynamic team. As a key member of our software development team, you will be responsible for designing and developing mobile applications that help patients manage their diabetes.Key ResponsibilitiesDevelop and maintain complex mobile applications (iOS or...


  • Los Angeles, California, United States Apposite Technologies LLC Full time

    Job Title: Senior Software Development EngineerJob Description:Apposite Technologies LLC is seeking a highly skilled Senior Software Development Engineer to join our team. The ideal candidate will have strong experience in DPDK, Docker, IP Networking, and Linux.We are looking for a talented engineer who can design and develop architectures for...


  • Los Angeles, California, United States Tik Tok Full time

    Job Title: Senior Machine Learning EngineerTikTok is a leading destination for short-form mobile video, and we're looking for a seasoned Senior Machine Learning Engineer to join our team. As a key member of our Brand Ads team, you will be responsible for creating innovative monetization products that drive engagement and revenue.Key Responsibilities:Create...


  • Los Angeles, California, United States Bitcoin Devs Company Full time

    Job OverviewWe're seeking a highly skilled Senior Backend Engineer to join our team at Bitcoin Devs Company. As a key member of our engineering team, you will design, build, and launch fraud risk products and features that drive business success.Key ResponsibilitiesDesign and manage distributed services that manage Marketplace fraud riskSolve complex...


  • Los Angeles, California, United States Bitcoin Devs Company Full time

    Job Title: Senior Backend EngineerJob DescriptionAs a key member of our team, you will play a crucial role in designing, building, and launching fraud risk products and features. Our services are built on ever-evolving infrastructure, and as a senior engineer, you will be responsible for scaling our tech stack to promote future evolvability.This is an...


  • Los Angeles, California, United States WerQ AI Full time

    Senior Software ArchitectWe are an innovative startup, WerQ AI, committed to transforming the workplace by focusing on its people. Our mission is to empower people with automated AI tools and enable them to leverage their networks for better opportunities and learning and development. We are currently seeking a highly skilled Senior Software Architect to...


  • Los Angeles, California, United States TechStarsGroup Full time

    Job Title: Senior Machine Learning EngineerWe are seeking a highly skilled Senior Machine Learning Engineer to join our data team. As a key member of our team, you will be responsible for designing, constructing, and maintaining machine learning pipelines and services.Key Responsibilities:Engineer and implement machine learning models into live...


  • Los Angeles, California, United States FreeForm Full time

    Job DescriptionWe are seeking a highly skilled Senior Software Engineer to join our team at Freeform. As a key member of our software development team, you will be responsible for designing and developing the print preparation software pipeline for our advanced production-scale metal 3D printing system.Your primary focus will be on automating CAD import,...


  • Los Angeles, California, United States Metropolis Full time

    About the RoleWe are seeking a Senior Software Engineer to join our team at Metropolis, an artificial intelligence company that uses computer vision technology to enable frictionless, checkout-free experiences in the real world.As a Senior Software Engineer, you will be responsible for designing and implementing our suite of parking products, working closely...


  • Los Angeles, California, United States Metropolis Full time

    Job OverviewMetropolis is seeking a highly skilled Senior Software Engineer to join our team. As a key member of our Engineering team, you will design and implement our suite of parking products, working closely with our product leaders and engineers to build an ecosystem of technologies at the intersection of parking, mobility, and real...


  • Los Angeles, California, United States Snap Full time

    About the RoleWe are seeking a highly skilled Staff Software Engineer to join our Machine Learning Feature Generation Team at Snap Inc. As a key member of our team, you will be responsible for driving technical direction and solving complex challenges in our ML infrastructure.Our team is responsible for building the declarative ML Feature Generation...


  • Los Angeles, California, United States Genentech Full time

    Transforming Drug Discovery with Machine LearningAt Genentech, we're pushing the boundaries of science to create a healthier future. Our Computational Sciences team is revolutionizing drug discovery with cutting-edge machine learning techniques. We're seeking a talented Senior Machine Learning Engineer to join our Prescient Design group and develop...