Senior JAX & AI Platform Architect
$320kNVIDIA Gruppe
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 include contributing to JAX, designing core components, and collaborating with researchers to enhance AI models. The position offers a competitive salary range of $320,000 to $488,750, along with equity and benefits. #J-18808-Ljbffr NVIDIA Gruppe
$272k - $431.25k
NVIDIA Gruppe is seeking experienced engineers for its Dynamo platform, focusing on scalable AI systems. You will develop the Kubernetes deployment, optimize GPU resource management, and work on intelligent routing and KV-cache management. Applicants should have 15+ years...PlatformSenior- NVIDIA Corporation is seeking a Senior HPC Architect to enhance GPU compute clusters. This role involves designing solutions for operationalizing NVIDIA products and collaborating closely with engineering teams. Ideal candidates should have over 8 years of experience in...PlatformSenior
$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...PlatformSenior- NVIDIA Gruppe in Santa Clara, California, is seeking an expert in AI training workloads to optimize performance on advanced hardware/software platforms. Candidates should have a PhD or MS in relevant fields plus substantial industry experience. This role involves deep learning...PlatformSenior
- NVIDIA Gruppe is looking for a Senior Deep Learning Computer Architect to design hardware accelerator architectures that enable cutting-edge AI algorithms. The role involves collaborating... ...performance on next-generation platforms. Candidates should hold an MS or PhD...PlatformSenior
$224k - $356.5k
NVIDIA Gruppe in Santa Clara is looking for a strong technical leader for its DriveOS software architecture group. The ideal candidate will have in-depth knowledge of complex systems, solid experience in Embedded Systems, and a Master's degree, with 12+ years of relevant...PlatformSeniorWork experience placement- NVIDIA Gruppe is looking for a senior test development engineer in Santa Clara, California... ...driving quality and automation for AI microservices, leading testing strategies... ...Python, and familiarity with cloud-native platforms such as Docker and Kubernetes. A diverse...PlatformSenior
$272k - $431.25k
JAX is the framework of choice for the most ambitious AI research, and the datacenters it runs on are evolving fast. NVIDIA... ...all of them. NVIDIA seeks a Senior Engineering Manager to define and... ...delivers peak performance across the platform, from single-accelerator...PlatformSenior$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... ...‑functional teams, principles and architects, coordinating effectively across organizational...PlatformSenior- ...next-generation computing experiences-from AI and data centers, to PCs, gaming and... ...We are seeking a Robotics AI Architect to define and scale next-generation Physical... ...systems, with a focus on complex robotic platforms (including humanoids). This role emphasizes...PlatformSenior
- ...Job Title: Senior Workato Integration Architect (MCP & AI exp) Location: Palo Alto, CA (3 days onsite) Duration: Long term contract... ...driven architectures on Workato , ensuring both platforms run in parallel during the transition (phased replacement...PlatformSeniorLong term contract
$184k - $287.5k
...Imagine shaping the future of AI and computing at one of the most innovative companies... .... We are seeking visionary computer architects to design and develop models for the next... ...of datacenter-scale accelerated computing platforms Evangelizing modern software architecture...PlatformSenior- GEICO is seeking a Distinguished Engineer to lead their CRM AI team in Palo Alto, California. This role involves driving the technical direction and execution of an AI-powered CRM roadmap, while fostering a high-performance culture. The ideal candidate will have extensive...PlatformSenior
- A leading AI solutions provider in California is seeking a Senior Simulation Engineer to design high-fidelity simulation platforms for advanced AI systems. The ideal candidate has 4+ years of experience in simulation-heavy product development, proficiency in C++ or Python...PlatformSenior
$224k - $356.5k
NVIDIA is seeking an AI Solutions Architect to join its Infrastructure Planning and Process Team NVIDIA is seeking an AI Solutions Architect... ...various operating systems (Windows/Linux/Android) and hardware platforms, including NVIDIA GPUs and Tegra processors. As an AI...PlatformSeniorWork experience placementWorldwide- A leading short-form video platform is seeking a Technical Product Designer in San Jose, California. This role involves leading design processes, implementing designs in code, and collaborating closely with Product and Engineering teams to enhance user experiences. The...PlatformSenior
- ...helping them shape their hybrid cloud and AI journeys. With support from our... ...About the Role We are seeking a Senior AI Architect - Enterprise Integrations to join our growing... ...enterprise systems including CRMs, ERPs, data platforms, and third-party APIs Establish a...PlatformSeniorWorldwide
- Advanced Micro Devices is seeking a Robotics AI Architect to lead the definition of next-generation Physical AI systems, focusing on complex robotic platforms. This role involves architectural leadership and collaboration with customers to achieve high-performance goals...PlatformSenior
$94.4k - $293.8k
Accenture is looking for an Advanced AI Architect in Mountain View, California, responsible for designing and delivering full-stack AI architecture for AI platforms. This role involves leading architecture workshops, designing multi-agent and governance solutions, and implementing...PlatformSenior- ...Senior Director, Design Engineering Req ID: 134544 Hiring Manager: Randy Clark... ...position is for a Senior Principal Engineer, AI/ML System Architect. As system architect,one will define... ...Define Complex AI/ML infrastructure platform solutions and lead a multi-...PlatformSeniorLocal areaRemote work
- Science Gateways is seeking a Principal Storage Architect & Team Lead to oversee a sophisticated academic storage ecosystem at Stanford... ...specialized team, ensuring the effective operation of storage platforms. Candidates must have over 10 years of experience in...PlatformSenior
$134.9k - $237.3k
...customers and employees. Through Genesys Cloud, the AI-powered Experience Orchestration platform, organizations can accelerate growth by delivering... ...that the job is tagged to) The Genesys AI Architect is a senior presales AI specialist who operates across three core...PlatformSeniorRemote workWork from homeWorldwideFlexible hours- ...Senior AI Architect Nexxa.ai is building artificial super intelligence for heavy industries — enabling machines, systems and operations to... ...with experience in production ML frameworks (e.g., PyTorch, JAX) Ability to independently scope ambiguous problems and deliver...Senior
$171.6k - $392.1k
...to build a better working world. ServiceNow – ServiceNow AI Architect Senior Manager In the digital economy, it takes more than good... ...in ServiceNow implementation experience and/or equivalent platform implementation experience Typically, no less than 2 - 5 years...PlatformSeniorSummer holidayWorldwideFlexible hours- ...FPGA Architect / Senior FPGA Engineer (Altera & Xilinx) We are looking for a seasoned FPGA expert with deep hands-on experience in Intel... ...Xilinx (UltraScale/UltraScale+) and Intel (Stratix, Agilex) platforms Perform IP mapping and integration (vendor IP + custom RTL...PlatformSenior
$171.6k - $392.1k
...help to build a better working world. ServiceNow - ServiceNow AI Architect Senior Manager In the digital economy, it takes more than good... ...experience in ServiceNow implementation experience and/or equivalent platform implementation experience Typically, no less than 2 - 5...PlatformSeniorSummer holidayWorldwideFlexible hours- ...Onsite or Hybrid) About Velaura AI Velaura AI is building the next generation of compute platforms for artificial intelligence. As... ...‑class team of researchers, architects, and engineers to rethink the foundations... ...ML frameworks such as PyTorch, JAX, or TensorFlow . Ability to...Platform
$228.9k - $412k
...market leader, bringing innovative AI-enhanced technology to over 8,... .... Our intelligent cloud-based platform seamlessly connects people,... ...Commercial segment. The Senior Director of Platform Architecture... ...Managers leading 60+ Platform Architects, cultivates C-suite...PlatformSeniorWork at officeRemote workFlexible hours$184k - $287.5k
We are now seeking a Senior Deep Learning Performance Architect! NVIDIA is looking for outstanding... ...performance modeling, and AI/deep learning to help... ...frameworks (e.g. Pytorch, JAX, TensorRT) Familiarity with... ...time, efficient computing platforms driving our success in this...PlatformSeniorFull timeWork experience placement$224k - $356.5k
...We are now looking for a Senior I/O Architect for our NVLink chip to chip team!NVIDIA is seeking an... ...trade off technical solutions to meet platform requirements for current and future products... ...an existing vacancy. NVIDIA uses AI tools in its recruiting processes....PlatformSenior
Do you want to receive more vacancies?
Subscribe and receive similar vacancies to Senior JAX & AI Platform Architect. Be the first to apply!
- senior game producer Santa Clara, CA
- senior manager process engineering Santa Clara, CA
- senior manufacturing engineer Santa Clara, CA
- senior manager clinical operations Santa Clara, CA
- senior optical engineer Santa Clara, CA
- senior lead project manager Santa Clara, CA
- senior manager quality engineering Santa Clara, CA
- senior device engineer Santa Clara, CA
- senior full stack developer Santa Clara, CA
- senior planner Santa Clara, CA

