Sign up to access all features of our service.
  • Job search
  • Favorites
  • Create a CV
    New
  • Salaries
  • Subscriptions

Engineering Leader, JAX Platform & AI Frameworks

$272k - $431.25k

NVIDIA Gruppe

NVIDIA Gruppe is looking for a Senior Engineering Manager based in Santa Clara, California. This role will challenge you to define and drive NVIDIA's JAX strategy while coordinating multiple teams to ensure peak performance across platforms. You will have a significant role in supporting next-generation architectures and bridging hardware capabilities with AI trends. Candidates should have strong software engineering and management experience, particularly in AI frameworks. The position offers a base salary range of 272,000 - 431,250 USD, equity, and benefits. #J-18808-Ljbffr NVIDIA Gruppe

Vacancy posted 1 day ago
Similar jobs that could be interesting for youBased on the Engineering Leader, JAX Platform & AI Frameworks in Santa Clara, CA vacancy
  • $20 - $71 per hour

     ...optimizations for deep learning frameworks using JAX, a powerful tool that...  ...fast, modular, and coordinated platform for a wide range of deep learning...  ...pushing the boundaries of AI and scientific computing....  ...Computer Science, Computer Engineering or an equivalent program area... 
    Platform
    Hourly pay
    Internship

    NVIDIA Gruppe

    Santa Clara, CA
    4 days ago
  • $184k - $356.5k

    NVIDIA Gruppe is seeking senior engineers to enhance its AI platform, focusing on performance optimizations in deep learning frameworks using JAX. Candidates should have strong programming and system design skills, along with proficiency in C/C++ and Python. The role involves... 
    Platform

    NVIDIA Gruppe

    Santa Clara, CA
    2 days ago
  •  ...seeking a Senior Manager, Observability Engineering to lead a team focused on building and scaling...  .... You will define strategy, ensure platform reliability, and collaborate with engineering...  ...ownership and strategic leadership for AI infrastructure. #J-18808-Ljbffr Dormont... 
    Platform

    Dormont Manufacturing Co

    Sunnyvale, CA
    2 days ago
  • NVIDIA Gruppe is seeking a Datacenter Product Development Engineering Leader in California to oversee the product lifecycle of advanced AI Supercomputing Platforms. You will work collaboratively across various engineering groups to launch innovative products that drive... 
    Platform

    NVIDIA Gruppe

    Santa Clara, CA
    2 days ago
  • $184k - $287.5k

    Overview 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...  ...with AI applied researchers and leaders to build future-proof models.... 
    Platform

    NVIDIA Gruppe

    Santa Clara, CA
    2 days ago
  • $272k - $431.25k

    JAX is the framework of choice for the most ambitious AI research, and the datacenters it runs on are evolving...  ...NVIDIA seeks a Senior Engineering Manager to define and...  ...performance across the platform, from single-...  ...— developing technical leaders and creating space for... 
    Platform

    NVIDIA Gruppe

    Santa Clara, CA
    2 days ago
  • $152k - $230k

     ...Product Marketing Manager to join their AI Platform Software team in Santa Clara,...  ...requires individuals passionate about AI frameworks and the ability to build go-to-market plans...  ...should have a MS/PhD in Computer Science or Engineering and at least 8 years of experience in a... 
    Platform

    NVIDIA

    Santa Clara, CA
    2 days ago
  • $184k - $356.5k

    NVIDIA Gruppe is seeking a Senior Engineer to lead the evolution of the core NIM Platform SDK and microservice framework in Santa Clara, California. This hands-on role involves...  ...contributing to production-grade software supporting AI applications. The position offers a... 
    Platform

    NVIDIA Gruppe

    Santa Clara, CA
    2 days ago
  • $272k - $431.25k

     ...team of customer-facing networking engineers. Provide organizational and...  ...deployment of current and next-generation AI systems, while preparing for future platforms. Engage directly with...  ...performance management. Develop future leaders by strengthening technical... 
    Platform

    NVIDIA

    Santa Clara, CA
    2 days ago
  • Ema, based in Palo Alto, is seeking an Engineering Leader to manage and scale multiple product lines in the Voice AI and Workforce Management sectors. You will drive the engineering organization responsible for building infrastructure, overseeing voice AI capabilities,... 
    Platform

    Ema

    Palo Alto, CA
    1 day ago
  •  ...tech firm in California is seeking a Head of Engineering to lead and scale their engineering team from...  ...year. This role demands a hands-on technical leader who values shipping velocity and possesses a strong background in AI technologies. Responsibilities include owning... 
    Platform

    Menlo Ventures

    Palo Alto, CA
    2 days ago
  • $183.8k - $263.6k

     .... Meet the Team Join our Frontend Engineering Team to shape and deliver the user...  ...Cloud Control—Cisco’s next-generation AI-powered cloud management platform. You’ll work in a modern...  ...Experience with automated testing frameworks (Jest, RTL, Playwright) and CI/CD pipelines... 
    Platform
    Full time
    Temporary work
    Work at office
    Local area
    Flexible hours

    Cisco Systems, Inc.

    Milpitas, CA
    2 days ago
  • $123.95k - $190.9k

     ...Responsibilities: Lead and manage a team of AI/ML engineers and data scientists, providing...  ...of AI/ML algorithms, techniques, and frameworks, with hands-on experience in model development...  ...-learn, etc. Experience with cloud platforms (e.g., AWS, Azure, GCP) and... 
    Platform
    Flexible hours

    Hewlett-Packard Company

    Palo Alto, CA
    2 days ago
  • Machine Learning, AI Software Engineer About The Role Help build AI powered product capabilities on a modern AI platform. You will work on model integration, retrieval systems, and evaluation frameworks that bring machine learning features into production. What You’ll... 
    Platform
    Permanent employment
    Contract work

    NLP PEOPLE

    Palo Alto, CA
    23 hours ago
  • $240k - $379.5k

     ...where they are on their AI journey on our GPUs -...  ...build best in class frameworks in OSS and support a robust...  ...Manager for AI Platform training you will be responsible...  ...also work with NVIDIA leaders to define clear...  ...Computer Science, Computer Engineering, or similar experience... 
    Platform

    NVIDIA Gruppe

    Santa Clara, CA
    2 days ago
  • 42dot is a mobility AI company committed to solving mobility challenges with software...  ...system. About the Role As an Android Engineer (UI Framework) at 42dot, you’ll be responsible for...  ...collaborate with design, HMI, system, and platform teams to deliver beautiful, intuitive,... 
    Platform

    42dot Inc.

    Sunnyvale, CA
    1 day ago
  •  ...communication libraries features in AI frameworks: from PoC to performance...  ...ultimate performance on NV platforms. Influence the roadmap of...  ...experience) with 5+ software engineering and HPC/AI experience...  ...Frameworks such as PyTorch, JAX, and inference engines such... 
    Platform

    NVIDIA Gruppe

    Santa Clara, CA
    2 days ago
  • $184k - $287.5k

    NVIDIA is the platform for every new AI-powered application. We seek a senior engineer to own and evolve the core NIM Platform SDK and microservice framework. This framework powers NVIDIA Inferencing Microservices (NIM). The ideal candidate has deep systems engineering... 
    Platform

    NVIDIA Gruppe

    Santa Clara, CA
    2 days ago
  • $229.9k - $262.4k

     ...creating responsible and reliable AI systems, changing banking for...  ...One has been an industry leader in using machine learning to...  ...world‑class applied science and engineering teams to deliver our industry...  ...customers. Our AI models and platforms empower teams across Capital... 
    Platform
    Full time
    Part time
    Local area

    Hobbsnews

    San Jose, CA
    3 days ago
  • $181.1k - $318.4k

    Senior Systems Framework Engineer, Vision Products Group Sunnyvale, California...  ...frameworks and APIs for our platforms. This engineer will enhance...  ...streaming, distributed AI workloads, and computer vision...  ...roadmap Act as a technical leader, fostering technical growth... 
    Platform
    Relocation

    Apple Inc.

    Sunnyvale, CA
    1 day ago
  •  ...money laundering. Our award-winning AI decision platform combines industry-leading unsupervised...  ...Role Summary We are seeking a Software Engineering Manager to lead our Platform...  ...with relational databases, SQL, and ORM frameworks (JPA, Hibernate) is a plus Experience... 
    Platform

    DataVisor

    Mountain View, CA
    3 days ago
  • Hewlett Packard Enterprise is seeking a Vice President of Platform Engineering located in Palo Alto, California. The role involves overseeing platform...  ..., driving platform engineering excellence, and enhancing AI-enabled experiences. The ideal candidate will have 15+ years... 
    Platform

    Hewlett Packard Enterprise

    Palo Alto, CA
    2 days ago
  •  ...Product Marketing Manager who specializes in AI frameworks for training and deployment. In Santa...  ...have an MS/PhD in Computer Science or Engineering, at least 8 years of experience in a technical...  ...in popular frameworks like PyTorch and JAX. Join us and make a lasting impact on AI... 
    Platform

    NVIDIA Corporation

    Santa Clara, CA
    2 days ago
  • $240k - $379.5k

    NVIDIA Gruppe is looking for a Product Manager for AI Platform training in Santa Clara, California. In this role, you will create products that help researchers and production model builders, develop strategies, and collaborate with customers. A degree in Computer Science... 
    Platform

    NVIDIA Gruppe

    Santa Clara, CA
    2 days ago
  •  ...will work closely with engineering to understand the technical...  ...leadership position in AI inference. What You’ll...  ...NVIDIA’s inference platform technical go-to-market...  ...on model architectures, frameworks, arXiv papers, whitepapers...  ...(PyTorch, TensorFlow, JAX), and inference‑... 
    Platform

    NVIDIA Gruppe

    Santa Clara, CA
    3 days ago
  • $224k - $356.5k

     ...Senior Deep Learning Software Engineer to design and build our...  ...developing features in high-level frameworks like PyTorch and JAX, designing and implementing...  ...a modular, scalable platform to seamlessly bridge training...  ...in Computer Science, AI, Applied Math, or related field... 
    Platform

    NVIDIA Gruppe

    Santa Clara, CA
    2 days ago
  • $152k - $241.5k

    Overview AI & Deep Learning Compiler Engineer for NVIDIA’s Deep Learning & AI Compiler (DLC) team. What you’ll...  ...with deep learning software framework teams and GPU architecture teams to...  ...and frameworks such as PyTorch and JAX. GPU kernel authoring and performance... 
    Platform

    NVIDIA Gruppe

    Santa Clara, CA
    2 days ago
  • Build and Deploy AI the right way, anywhere. The...  ...Compute Infrastructure Platform provides an "end-to-end...  ...As a Staff AI Runtime Engineer , you’ll play a pivotal...  ...Build Internal Tooling & Frameworks: Design and maintain libraries...  .... PyTorch, TensorFlow, JAX) for large‑scale... 
    Platform
    Work at office

    FlexAI

    Santa Clara, CA
    4 days ago
  • $168.8k - $241.2k

     ...The Common Hardware Group (CHG) creates innovative hardware platforms central to the AI era, powering Cisco’s core Switching, Routing, and...  ...future of connectivity! Your Impact You will work among engineers as a direct contributor, designing and operating our growing... 
    Platform
    Full time
    Temporary work
    Local area
    Worldwide
    Flexible hours

    Cisco Systems, Inc.

    Milpitas, CA
    2 days ago
  • $224k - $356.5k

    ## Software Engineering Manager, Robotics Neural Reconstruction and...  ...the future of physical AI!**What you’ll be doing:***...  ...Isaac teams including Sim/Lab, Platform and SQA to deploy, validate...  ...experience with Deep Learning frameworks (PyTorch, JAX, TensorFlow) and physics... 
    Platform

    NVIDIA Corporation

    Santa Clara, CA
    1 day ago

Do you want to receive more vacancies?

Subscribe and receive similar vacancies to Engineering Leader, JAX Platform & AI Frameworks. Be the first to apply!