jax Jobs in san jose, ca

Refine Results
41 - 58 of 58 Jobs

Senior High-Performance LLM Training Engineer

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

Machine Learning Engineer, Efficiency Engineering - USDS

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

Sr. Worldwide Specialist, GenAI, Initiatives

Amazon Web Services, Inc.

Santa Clara, California, USA

Full-time

Do you want to help define the future of Go to Market (GTM) at AWS using generative AI (GenAI)? You will be part of the core worldwide GenAI Training and Inference team, responsible for defining, building, and deploying targeted strategies to accelerate customer adoption of our services and solutions across industry verticals. You will be working directly with the most important customers (across segments) in the GenAI model training and inference space helping them adopt and scale large-scale w

Staff AI/ML Engineer - Onboard Embodied AI

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)

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++/Python

AMD (Advanced Micro Devices)

San Jose, 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

Principal AI Engineer, Enterprise AI Platform

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 APIs

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, Optimus

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)

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 excellen

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

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 & Robotics

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

Staff Software Engineer, Machine Learning

Discord Inc.

Remote

Full-time

Discord is used by over 200 million people every month for many different reasons, but there's one thing that nearly everyone does on our platform: play video games. Over 90% of our users play games, spending a combined 1.5 billion hours playing thousands of unique titles on Discord each month. Discord plays a uniquely important role in the future of gaming. We are focused on making it easier and more fun for people to talk and hang out before, during, and after playing games. We're currently l

Senior Software Engineer - Python Numerical Computing Libraries

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)

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

Senior Machine Learning Engineer

Randstad - Torc

Remote

Contract

We are looking for a Senior Machine Learning Engineer to join our team and drive the development of advanced AI-driven solutions. You will be responsible for designing, deploying, and optimizing machine learning models that power real-world applications. This role requires deep expertise in model architecture, data engineering, MLOps, and production-scale AI systems. What you'll do:Design, develop, and deploy scalable, production-ready machine learning models for real-world applications.Architec

Lead Java Middle Tier Developer- C13 - JERSEY CITY

Citi

Remote or Jersey City, New Jersey, USA

Full-time

The Lead Java Middle Tier Developer is a senior level position responsible for establishing and implementing new or revised application systems and programs in coordination with the Technology team. The overall objective of this role is to lead applications systems analysis and programming activities. Responsibilities: Write good and well-tested code in JavaSOA API middleware implementation (both REST and SOAP)Write user/supporter documentation Development environment support (mostly automation

Technical Sales Manager - MSP Experience - Territory - South Florida, Orlando, Jacksonville, Tallahassee and Atlanta

Stone Search

Remote

Full-time

Territory- South, specifically South FL, Orlando, Jax, Tallahassee, and Atlanta Work from your home office As a Technical Sales Account Manager in the Information Technology and Services industry, you will play a crucial role in driving revenue growth and building strong relationships with clients. Having experience in Managed Services, In this dynamic position, you will be responsible for understanding clients' technical needs, presenting solutions, and closing sales deals. You will work closel