jax Jobs in california

Refine Results
61 - 80 of 90 Jobs

GPU SME- 10+ yrs- Remote (US Only)

iMedhas Consulting Services

•

Remote

Third Party, Contract

Job description: GPU / TPU Experience Familiarity with hands on IaC is a must Terraform GKE Networking Storage Python Library familiarity with: numpy/pandas/Pytorch/JAX, including optimization Experience with Nvidia and/or Google TPU hardware in GCE and GKE ML specific Google Cloud Platform products: Parallel store, Hyperdisk ML, TCP Direct Troubleshooting and optimization of large (1000s of nodes) GKE clusters
Easy Apply
Posted 2 days ago | Updated 1 day ago

GPU SME- 10+ yrs- Remote (US Only)

iMedhas Consulting Services

•

Remote

Third Party, Contract

Job description: GPU / TPU Experience Familiarity with hands on IaC is a must Terraform GKE Networking Storage Python Library familiarity with: numpy/pandas/Pytorch/JAX, including optimization Experience with Nvidia and/or Google TPU hardware in GCE and GKE ML specific Google Cloud Platform products: Parallel store, Hyperdisk ML, TCP Direct Troubleshooting and optimization of large (1000s of nodes) GKE clusters
Easy Apply
Posted 2 days ago

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
Posted 60+ days ago | Updated 4 hours ago

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,
Posted 60+ days ago | Updated 1 hour ago

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
Posted 56 days ago | Updated 4 hours ago

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
Posted 60+ days ago | Updated 4 hours ago

Senior Software Engineer, AI Resiliency

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
Posted 60+ days ago | Updated 4 hours ago

Senior MLOps Engineer, GenAI Framework

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
Posted 14 days ago | Updated 4 hours ago

Senior Director of Data Engineering

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
Posted 2 days ago | Updated 4 hours ago

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
Posted 28 days ago | Updated 4 hours ago

Senior, Data Scientist | Gen AI Vision

Walmart Inc.

•

Sunnyvale, California, USA

Full-time

Position Summary What you'll do Senior Data Scientist (Gen AI, Computer Vision) Join Walmart and your work could help over 275 million global customers live better every week. Yes, we are the Fortune #1 company. But you'll quickly find we're a company who wants you to feel comfortable bringing your whole self to work. A career at Walmart is where the world's most complex challenges meet a kinder way of life. Our mission spreads far beyond the walls of our stores. Join us and you'll discover w
Posted 6 days ago | Updated 4 hours ago

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
Posted 60+ days ago | Updated 4 hours ago

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
Posted 60+ days ago | Updated 4 hours ago

Senior Research Engineer for Reinforcement Learning

NVIDIA Corporation

•

Santa Clara, California, USA

Full-time

NVIDIA is searching for a senior or principal engineer who specializes in large-scale reinforcement learning and policy learning in the Generalist Embodied Agent Research (GEAR) group. Our team is leading Project GR00T, NVIDIA's moonshot initiative at building foundation models and full-stack technology for humanoid robots. You will work with an amazing and collaborative research team that consistently produces influential works on multimodal foundation models, large-scale robot learning, embodi
Posted 60+ days ago | Updated 4 hours ago

AI Engineer, Data Scaling, Self-Driving

Tesla Motors

•

Palo Alto, California, USA

Full-time

At Tesla, you will have access to unparalleled resources that set us apart from other companies in the AI industry. With the world's largest self-driving dataset, you will have a unique opportunity to develop and optimize a large-scale data engine that powers our autonomous driving systems. Tesla's extensive fleet of vehicles generates massive amounts of real-world data daily, giving you unprecedented access to diverse driving scenarios and edge cases. Additionally, Tesla offers one of the highe
Posted 60+ days ago | Updated 4 hours ago

Senior Research Engineer, Foundation Model Training Infrastructure

NVIDIA Corporation

•

Santa Clara, California, USA

Full-time

NVIDIA is searching for a senior or principal engineer who specializes in building cutting-edge infrastructure for large-scale foundation model training in the Generalist Embodied Agent Research (GEAR) group. Our team is leading Project GR00T, NVIDIA's moonshot initiative at building foundation models and full-stack technology for humanoid robots. You will work with an amazing and collaborative research team that consistently produces influential works on multimodal foundation models, large-sca
Posted 60+ days ago | Updated 4 hours ago

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
Posted 57 days ago | Updated 4 hours ago

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
Posted 39 days ago | Updated 4 hours ago

Technical Marketing Engineer - AI Platform Software

NVIDIA

•

Santa Clara, California, USA

Full-time

NVIDIA software powers today's breakthroughs in AI! To enable researchers and developers to keep pace with this dynamic field, we seek a technical marketing expert who understands the AI platform software stack and the ecosystem! This role will craft the fundamental technical content educating developers how to write software's using NVIDIA's AI platforms through technical blog posts, user guides, walk-throughs, benchmark and more. This material is essential in mentoring developers about the lat
Posted 60+ days ago | Updated 2 hours ago

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
Posted 60+ days ago | Updated 4 hours ago