JAX Jobs in San Jose, CA

41 - 60 of 64 Jobs

Principal Deep Learning Software Engineer, LLM PerformanceJob Title - Principal Deep Learning Software Engineer, LLM Performance

NVIDIA CorporationCompany Name - NVIDIA Corporation

•

Remote or Santa Clara, California, USA

Full-time

We are now looking for a Principal Deep Learning Software Engineer, LLM Performance! NVIDIA is seeking an experienced Deep Learning Engineer passionate about analyzing and improving the performance of LLM inference! NVIDIA is rapidly growing our research and development for Deep Learning Inference and is seeking excellent Software Engineers at all levels of expertise to join our team. Companies around the world are using NVIDIA GPUs to power a revolution in deep learning, enabling breakthroughs

Senior High-Performance LLM Training EngineerJob Title - Senior High-Performance LLM Training Engineer

NVIDIA CorporationCompany Name - NVIDIA Corporation

•

Santa Clara, California, USA

Full-time

We are now looking for a Senior High-Performance LLM Training Engineer! NVIDIA is seeking experienced engineers specializing in performance analysis and optimization to improve the efficiency of LLM training workloads, which are shaping the world's most advanced computing systems. This position focuses on optimizing NVIDIA's high-performance LLM software stack in frameworks like PyTorch and JAX for high-performance training on thousands of GPUs, while also helping shape hardware roadmaps for th

AI and ML Infra Software Engineer, GPU ClustersJob Title - AI and ML Infra Software Engineer, GPU Clusters

NVIDIA CorporationCompany Name - NVIDIA Corporation

•

Santa Clara, California, USA

Full-time

NVIDIA has been transforming computer graphics, PC gaming, and accelerated computing for more than 25 years. It's a unique legacy of innovation that's fueled by great technology-and amazing people. Today, we're tapping into the unlimited potential of AI to define the next era of computing. An era in which our GPU acts as the brains of computers, robots, and self-driving cars that can understand the world. Doing what's never been done before takes vision, innovation, and the world's best talent.

Lead Research EngineerJob Title - Lead Research Engineer

Aduril IndustriesCompany Name - Aduril Industries

•

Mountain View, California, USA

Full-time

Anduril Industries is a defense technology company with a mission to transform U.S. and allied military capabilities with advanced technology. By bringing the expertise, technology, and business model of the 21st century's most innovative companies to the defense industry, Anduril is changing how military systems are designed, built and sold. Anduril's family of systems is powered by Lattice OS, an AI-powered operating system that turns thousands of data streams into a realtime, 3D command and c

Machine Learning Engineer, Efficiency Engineering - USDSJob Title - Machine Learning Engineer, Efficiency Engineering - USDS

TikTokCompany Name - TikTok

•

San Jose, California, USA

Full-time

Location : San Jose Employment Type : Regular Job Code : A87619 Apply to this job Share this listing: Responsibilities The Efficiency Engineering team is all about our passion for crafting innovative tools and applications that empower IT operations and devops teams to achieve new levels of efficiency. We're a tight-knit crew of experienced developers, engineers and problem solvers fueled by a shared vision: streamlining operations, reducing manual workload, and empowering teams to do the

Staff AI/ML Engineer - Onboard Embodied AIJob Title - Staff AI/ML Engineer - Onboard Embodied AI

General MotorsCompany Name - General Motors

•

Mountain View, California, USA

Full-time

Job Description Hybrid: This role is categorized as hybrid. This means the successful candidate is expected to report to the Mountain View Technical Center in the Bay Area three times per week, at minimum. Role: As a Technical Lead in Machine Learning within the Onboard Embodied AI organization, you will be a senior individual contributor driving cutting-edge end-to-end machine learning solutions directly impacting autonomous driving performance. Your role is pivotal in designing, architecting,

Principal Solutions Engineer, Infrastructure (SLURM & AI Focus)Job Title - Principal Solutions Engineer, Infrastructure (SLURM & AI Focus)

AMD (Advanced Micro Devices)Company Name - AMD (Advanced Micro Devices)

•

Santa Clara, California, USA

Full-time

WHAT YOU DO AT AMD CHANGES EVERYTHING We care deeply about transforming lives with AMD technology to enrich our industry, our communities, and the world. Our mission is to build great products that accelerate next-generation computing experiences - the building blocks for the data center, artificial intelligence, PCs, gaming and embedded. Underpinning our mission is the AMD culture. We push the limits of innovation to solve the world's most important challenges. We strive for execution excellenc

Staff Software Development Engineer - AI/ML frameworks, C/C++/PythonJob Title - Staff Software Development Engineer - AI/ML frameworks, C/C++/Python

AMD (Advanced Micro Devices)Company Name - AMD (Advanced Micro Devices)

•

San Jose, California, USA

Full-time

Senior Software Engineer, AI ResiliencyJob Title - Senior Software Engineer, AI Resiliency

NVIDIA CorporationCompany Name - NVIDIA Corporation

•

Santa Clara, California, USA

Full-time

We are now looking for a Senior Software Engineer for AI Resiliency. At NVIDIA, we are pushing the boundaries of what's possible in AI. We are currently seeking a Senior Software Engineer to lead the development of AI software resiliency for the most powerful AI supercomputers in the world. As a member of our AI Software Resiliency team, you will play a pivotal role in defining and implementing critical resiliency features for AI supercomputers at a scale of 100,000+ GPUs. Your expertise will b

Senior MLOps Engineer, GenAI FrameworkJob Title - Senior MLOps Engineer, GenAI Framework

NVIDIA CorporationCompany Name - NVIDIA Corporation

•

Santa Clara, California, USA

Full-time

NVIDIA is looking for a dedicated and motivated senior build and continuous integration (CI/CD) engineer for its GenAI Frameworks (Megatron-LM and NeMo Framework) team. Megatron-LM and NeMo Framework are open-source, scalable and cloud-native frameworks built for researchers and developers working on Large Language Models (LLM), Multimodal (MM), and Video Generation. Megatron-LM and NeMo Framework provide end-to-end model training, including data curation, alignment, customization, evaluation, d

Principal AI Engineer, Enterprise AI PlatformJob Title - Principal AI Engineer, Enterprise AI Platform

PaloAlto NetworksCompany Name - PaloAlto Networks

•

Santa Clara, California, USA

Full-time

Company Description Our Mission At Palo Alto Networks everything starts and ends with our mission: Being the cybersecurity partner of choice, protecting our digital way of life. Our vision is a world where each day is safer and more secure than the one before. We are a company built on the foundation of challenging and disrupting the way things are done, and we're looking for innovators who are as committed to shaping the future of cybersecurity as we are. Who We Are We take our mission of

Senior Math Libraries Engineers - Python APIsJob Title - Senior Math Libraries Engineers - Python APIs

NVIDIA CorporationCompany Name - NVIDIA Corporation

•

Remote or Santa Clara, California, USA

Full-time

NVIDIA is looking for a self-motivated and specialist software engineer for the design and development of high-performant Python APIs for our math libraries. In the last decade, Python has become the de-facto leading programming language for engineers in AI and data science, and more recently in HPC and scientific computing. NVIDIA has been at the forefront of providing GPU accelerated Deep Learning frameworks. These frameworks provide an efficient high-level programming interface allowing their

Deep Learning Vision Engineer, OptimusJob Title - Deep Learning Vision Engineer, Optimus

Tesla MotorsCompany Name - Tesla Motors

•

Palo Alto, California, USA

Full-time

Tesla is on a path to build humanoid robots at scale to automate repetitive and boring tasks. The goal of our Machine Learning team is to build and demonstrate a general robot learning system that can leverage AI to perform complex physical tasks, ranging from full body locomotion, and precise manipulation This specific role will do hands-on data analysis for Optimus's Machine Learning team, as well as build data pipelines, tools and applications to automate those analysis. This is a unique oppo

Machine Learning Software Engineer (SMTS)Job Title - Machine Learning Software Engineer (SMTS)

AMD (Advanced Micro Devices)Company Name - AMD (Advanced Micro Devices)

•

Santa Clara, California, USA

Full-time

Senior Director, Data EngineeringJob Title - Senior Director, Data Engineering

Visa Inc.Company Name - Visa Inc.

•

Foster City, California, USA

Full-time

Company Description Visa is a world leader in payments and technology, with over 259 billion payments transactions flowing safely between consumers, merchants, financial institutions, and government entities in more than 200 countries and territories each year. Our mission is to connect the world through the most innovative, convenient, reliable, and secure payments network, enabling individuals, businesses, and economies to thrive while driven by a common purpose - to uplift everyone, everywhe

ML Data Engineer - Healthcare Data Curation & Cleaning (1 Year Fixed Term)Job Title - ML Data Engineer - Healthcare Data Curation & Cleaning (1 Year Fixed Term)

Stanford UniversityCompany Name - Stanford University

•

Stanford, California, USA

Full-time

Stanford University is seeking a Big Data Architect 1 for a 1 year fixed term (possibility of renewal) to design and develop applications, test and build automation tools and support the development of Big Data architecture and analytical solutions. About Us: The Department of Biomedical Data Science merges the disciplines of biomedical informatics, biostatistics, computer science and advances in AI. The intersection of these disciplines is applied to precision health, leveraging data across th

AI Research Engineer, Foundation Models, Autonomy & RoboticsJob Title - AI Research Engineer, Foundation Models, Autonomy & Robotics

Tesla MotorsCompany Name - Tesla Motors

•

Palo Alto, California, USA

Full-time

Join us to help build the vision and language models that understand the world and act intelligently in it. We are hiring exceptional Software Engineers that can work across the machine learning stack, from designing and training neural architectures to deciding what data to collect and how to evaluate the model. You will join a small group of world-class deep learning experts to develop novel state-of-the-art neural networks and push the boundaries of AI research and development. Your work will

Senior Software Engineer - Python Numerical Computing LibrariesJob Title - Senior Software Engineer - Python Numerical Computing Libraries

NVIDIA CorporationCompany Name - NVIDIA Corporation

•

Remote or Santa Clara, California, USA

Full-time

We are looking for an experienced software professional to contribute to design and development of accelerated and distributed implementations of Python APIs for numerical computing. In the last decade, Python has become the de-facto programming language for practitioners in AI, data science and HPC, through popular frameworks such as NumPy, SciPy, TensorFlow and PyTorch. These frameworks provide an efficient high-level programming interface, allowing their users to focus on their application wh

Machine Learning Systems Engineer (1 Year Fixed Term)Job Title - Machine Learning Systems Engineer (1 Year Fixed Term)

Stanford UniversityCompany Name - Stanford University

•

Stanford, California, USA

Full-time

The Department of Ophthalmology in the School of Medicine at Stanford University is launching an interdisciplinary Neuro-AI project dedicated to building a foundation model of the brain. This endeavor will involve multiple labs and faculty across the Stanford campus, including the Wu Tsai Neurosciences Institute, Stanford Bio-X, and the Human-Centered Artificial Intelligence Institute. Leveraging cutting-edge advances in electrophysiology and machine learning, this project aims to create a funct

Lead Data Scientist, NLP & Strategic Member InsightsJob Title - Lead Data Scientist, NLP & Strategic Member Insights

CVS HealthCompany Name - CVS Health

•

Connecticut, USA

Full-time

At CVS Health, we're building a world of health around every consumer and surrounding ourselves with dedicated colleagues who are passionate about transforming health care. As the nation's leading health solutions company, we reach millions of Americans through our local presence, digital channels and more than 300,000 purpose-driven colleagues - caring for people where, when and how they choose in a way that is uniquely more connected, more convenient and more compassionate. And we do it all w