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
$320k
...NVIDIA Gruppe in Santa Clara is looking for a distinguished engineer to develop its AI platform, focusing on deep learning frameworks like JAX. This role requires strong programming and system design skills, along with excellent communication capabilities. Responsibilities...Platform$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...Platform
$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$272k - $431.25k
...JAX is the framework of choice for the most ambitious AI research, and the datacenters it runs on are... ...NVIDIA seeks a Senior Engineering Manager to define and drive... ...performance across the platform, from single-... ...— developing technical leaders and creating space for...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- 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 - $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$152k - $230k
NVIDIA AI is looking for a technical product marketing manager passionate about AI frameworks to join its AI Platform Software team in Santa Clara, California. The role involves building marketing plans, creating supportive materials, and representing NVIDIA at events....Platform- Cacheflow is seeking a Sr. Engineering Manager for the App Framework team in Mountain View. In this role, you will build foundational service frameworks for microservices at Databricks, ensuring fast, reliable, and secure applications that benefit thousands of engineers...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- ...customers where they are on their AI journey on our GPUs, building best‑in‑class frameworks in OSS and supporting a robust... .... As a Product Manager for AI Platform Recommender Systems, you will... ...in Computer Science, Computer Engineering, or similar experience (or equivalent...Platform
$183.8k - $263.6k
...office. Meet the Team Join our Frontend Engineering Team to shape and deliver the user... ...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- ...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
$197.3k - $313.7k
...within the Agentforce platform that are used by millions... ..., product owners, engineers, user experience designers... ...strategies, design automation frameworks, write unit and... ...and act as a Technical Leader for teams Adopt, embrace... ...Experience with modern AI/ML frameworks, systems,...PlatformFlexible hours$224.92k
...Lead a high-performing software engineering team to architect, develop, and deliver AI-driven, scalable enterprise solutions... ...cloud-based file management platforms, real-time collaboration tools,... ...CD pipelines, automated testing frameworks, and performance tuning processes...PlatformWork at officeRemote work$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$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- ...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
$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- ...Meet the Team The Cisco AI Software & Platform Group incubates and delivers Generative AI based... ...experienced and innovative ML Operations Engineer to join our global DevOps team. In... ...such as Terraform, Helm, and GitOps frameworks. Monitor models in production for performance...PlatformTemporary work
$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$212.3k - $275.8k
...received . Meet the Team The Cisco AI Software & Platform Group incubates and delivers... ...highly experienced and innovative DevOps Engineer to join our global DevOps team. In this... ...such as Terraform, Helm, and GitOps frameworks. Continuously evaluate and adopt...PlatformFull timeTemporary workLocal areaFlexible hours- 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
$152k - $241.5k
...We are seeking a Senior Software Engineer to drive integration of the NVIDIA Grove project... ...across a set of leading open-source AI frameworks. In this role, you will develop production... ...inference, or multi‑tenant AI platforms. Experience building SDKs/APIs or developer...Platform- ...generation computing experiences-from AI and data centers, to PCs,... ...member of the LLM inference framework team, you will be responsible... ...deployments and benchmarking platforms (e.g., InferenceMax, MLPerf,... ...intersection of inference engines, distributed systems, and GPU...Platform
- ...NVIDIA Gruppe is seeking a Product Manager for AI Platform Recommender Systems in Santa Clara, California. This role focuses on creating innovative products that enhance RecSys and Generative Recommendation Models. The ideal candidate will have over 10 years of experience...Platform
- ...Hewlett Packard Enterprise is seeking a Vice President of Platform Engineering located in Palo Alto, California. The role involves overseeing... ...products, driving platform engineering excellence, and enhancing AI-enabled experiences. The ideal candidate will have 15+ years...Platform
$221k - $320k
...Adobe is looking for a Senior Engineering Manager in San Jose, CA. You will lead the development of a central AI platform and mentor a dynamic engineering team. The ideal candidate has over 10 years of engineering experience and proven success in AI platforms. The role...Platform- ...Capital One in San Jose is seeking a Sr. Manager, AI Engineering to lead innovative AI initiatives. The role involves collaborating across teams to develop AI-powered products that transform banking. The ideal candidate will possess extensive experience in AI and ML,...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
- platform engineer Santa Clara, CA
- platform engineering manager Santa Clara, CA
- data platform engineer Santa Clara, CA
- client platform engineer Santa Clara, CA
- senior platform engineer Santa Clara, CA
- platform product manager Santa Clara, CA
- platform manager Santa Clara, CA

