jax Jobs in san francisco, ca

1 - 20 of 22 Jobs

Software Engineer, ML Inference Compiler & Deployment, GPU, CPUJob Title - Software Engineer, ML Inference Compiler & Deployment, GPU, CPU

Tesla MotorsCompany Name - Tesla Motors

•

Palo Alto, California, USA

Full-time

As a Software Engineer within our Autonomy teams, you will contribute to one of the most advanced and widely deployed AI Platforms in the world for Autopilot and our Humanoid Robot, Optimus. In this role, you will be responsible for the internal working of the AI inference stack and compiler running neural networks in millions of Tesla vehicles and Optimus. You will collaborate closely with the AI Engineers and Hardware Engineers to understand the full inference stack and design the compiler to

Software Engineer, ML Inference Compiler & Deployment, AI FrameworksJob Title - Software Engineer, ML Inference Compiler & Deployment, AI Frameworks

Tesla MotorsCompany Name - Tesla Motors

•

Palo Alto, California, USA

Full-time

As a Software Engineer within our Autonomy teams, you will contribute to one of the most advanced and widely deployed AI Platforms in the world, powering Autopilot and our Humanoid Robot, Optimus. In this role, you will be responsible for the internal working of the AI inference stack and compiler running neural networks in millions of Tesla vehicles and Optimus. You will collaborate closely with the AI Engineers and Hardware Engineers to understand the full inference stack and design the compi

Software Engineer, ML Inference Compiler & Deployment, AI AcceleratorJob Title - Software Engineer, ML Inference Compiler & Deployment, AI Accelerator

Tesla MotorsCompany Name - Tesla Motors

•

Palo Alto, California, USA

Full-time

Machine Learning Kernel Performance Engineer, DojoJob Title - Machine Learning Kernel Performance Engineer, Dojo

Tesla MotorsCompany Name - Tesla Motors

•

Palo Alto, California, USA

Full-time

As a member of the Dojo Machine Learning software team, you will be responsible for developing and optimizing machine learning kernels to run on a massively parallel machine. The ideal candidate will have a strong background in kernel development and performance optimization for AI workloads, with a passion for delivering high-performance implementations, which are simple to use. Responsibilities Develop and validate datapath kernels on a massively parallel machineDesign and deliver implementat

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

Developer Technologies Engineer, Robotics Reinforcement LearningJob Title - Developer Technologies Engineer, Robotics Reinforcement Learning

NVIDIA CorporationCompany Name - NVIDIA Corporation

•

Remote or Santa Clara, California, USA

Full-time

Today, NVIDIA is 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. As an NVIDIAN, you'll be immersed in a diverse, encouraging environment where everyone is inspired to do their best work. Come join the team and see how we can make a lasting impact on the world

Senior GenAI Algorithms EngineerJob Title - Senior GenAI Algorithms Engineer

NVIDIA CorporationCompany Name - NVIDIA Corporation

•

Remote or Santa Clara, California, USA

Full-time

We are now looking for a Senior Gen AI Algorithms Engineer! NVIDIA is seeking engineers to design, develop and optimize Artificial Intelligence solutions to diverse real-world problems. If you have a strong understanding of deep learning and in particular large language models and their multimodal variants, then this role may be a great fit for you! Collaborate and interact with internal partners, users, and members of the open source community to analyze, define and implement highly optimized A

Principal Software Engineer, TensorRT-LLMJob Title - Principal Software Engineer, TensorRT-LLM

NVIDIA CorporationCompany Name - NVIDIA Corporation

•

Remote or Santa Clara, California, USA

Full-time

We are now looking for a Principal Software Engineer, TensorRT-LLM ! NVIDIA is hiring experienced principal software engineer for its TensorRT-LLM team. Academic and commercial groups around the world are using GPUs to power a revolution in AI, enabling breakthroughs in areas like content creation, code generation and information synthesis, that have put AI on the precipice of an "iPhone moment". Join the team building the AI serving software which is foundational to product lines within NVIDIA

Staff Software Engineer, Machine LearningJob Title - Staff Software Engineer, Machine Learning

Discord Inc.Company Name - 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

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 Deep Learning Software Engineer, Inference and Model OptimizationJob Title - Senior Deep Learning Software Engineer, Inference and Model Optimization

NVIDIA CorporationCompany Name - NVIDIA Corporation

•

Remote or Santa Clara, California, USA

Full-time

NVIDIA is at the forefront of the generative AI revolution! The Algorithmic Model Optimization Team specifically focuses on optimizing generative AI models such as large language models (LLM) and diffusion models for maximal inference efficiency using techniques ranging from neural architecture search and pruning to sparsity, quantization, and automated deployment strategies. Our work includes conducting applied research to improve model efficiency as well as developing an innovative software pl

Senior Deep Learning Software Engineer, LLM PerformanceJob Title - Senior 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 Senior 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 in

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

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

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

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

Senior Machine Learning EngineerJob Title - Senior Machine Learning Engineer

Randstad - TorcCompany Name - 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

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

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

Lead Java Middle Tier Developer- C13 - JERSEY CITYJob Title - Lead Java Middle Tier Developer- C13 - JERSEY CITY

CitiCompany Name - 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