Engineering Leader, JAX Platform & AI Frameworks
$272k - $431.25kNVIDIA 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
$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...PlatformHourly payInternship$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- ...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
- 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
$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$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$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$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$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- 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
- ...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
$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...PlatformFull timeTemporary workWork at officeLocal areaFlexible hours$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...PlatformFlexible hours- 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...PlatformPermanent employmentContract work
$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- 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
- ...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
$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$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...PlatformFull timePart timeLocal area$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...PlatformRelocation- ...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
- 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
- ...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
$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- ...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
$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$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- 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...PlatformWork at office
$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...PlatformFull timeTemporary workLocal areaWorldwideFlexible hours$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
Do you want to receive more vacancies?
Subscribe and receive similar vacancies to Engineering Leader, JAX Platform & AI Frameworks. Be the first to apply!
- engineering leader Santa Clara, CA
- engineering team lead Santa Clara, CA
- platform developer Santa Clara, CA
- senior platform engineer Santa Clara, CA
- platform engineering manager Santa Clara, CA
- platform engineer Santa Clara, CA
- data platform engineer Santa Clara, CA
- platform product manager Santa Clara, CA
- platform manager Santa Clara, CA
- manufacturing engineering team leader

