JAX Jobs in San Francisco, CA

Refine Results
1 - 20 of 23 Jobs

Business Unit Manager - SVP (Enterprise Data Office)(JAX, FL)(Irving, TX)

Citi

Remote or Jacksonville, Florida, USA

Full-time

About Citi Citi, the leading global bank, has approximately 200 million customer accounts and does business in more than 160 countries and jurisdictions. Citi provides consumers, corporations, governments and institutions with a broad range of financial products and services, including consumer banking and credit, corporate and investment banking, securities brokerage, transaction services, and wealth management. Our core activities are safeguarding assets, lending money, making payments and ac

IT Loan IQ Business Lead Analyst - VP(Irving TX, JAX FL, New Castle, DE)

Citi

Remote or Irving, Texas, USA

Full-time

The IT Loan IQ Business Lead Analyst is a senior-level position responsible for liaising between business users and technologists to exchange information in a concise, logical and understandable way in coordination with the Technology team. The overall objective of this role is to contribute to continuous iterative exploration and investigation of business performance and other measures to gain insight and drive business planning. Responsibilities: Provide input during development and implement

Sr Machine Learning Engineer

Uber Corporate

San Francisco, California, USA

Full-time

About the Teams: Coordinated Structural Pricing: The coordinated structural pricing and simulator team works on setting rider and driver prices to achieve marketplace balance. We maintain many offline models to simulate and optimize marketplace price levels at a coarse granularity to achieve growth in trips and GB while maintaining marketplace reliability and VC neutrality. We are actively working on a new pricing and simulation tech called Pricer-Planner and exploring many different avenues fo

Distinguished Engineer, Machine Learning (Modeling and Tuning)

Cisco Systems, Inc.

San Francisco, California, USA

Full-time

Who We Are Cisco ThousandEyes is a Digital Experience Assurance platform that empowers organizations to deliver flawless digital experiences across every network - even the ones they don't own. Powered by AI and an unmatched set of cloud, internet and enterprise network telemetry data, ThousandEyes enables IT teams to proactively detect, diagnose, and remediate issues - before they impact end- user experiences. ThousandEyes is deeply integrated across the entire Cisco technology portfolio and

Sr. Software Engineer, PyTorch Export, AI Inference

Tesla Motors

Palo Alto, California, USA

Full-time

Our AI Inference team puts ML models into production - we train and deploy large neural networks for efficient inference on compute-constrained edge devices (CPU / GPU / AI ASIC). The nature of this role is multi-disciplinary - you will work at the intersection of machine learning and systems by building the ML frameworks and infrastructure that enable the seamless deployment, and inference of all neural networks that run on Autopilot and Optimus. Responsibilities Build robust AI frameworks to

Staff Software Engineer, ML Inference Compiler & Deployment, Optimus

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 running neural networks in Optimus and millions of Tesla vehicles. You will collaborate closely with the Optimus AI Engineers and AI Hardware Engineers to understand the full inference stack, co-design models to fit the

AI System Research and Development Engineer - Optimization

Snowflake Inc.

Menlo Park, California, USA

Full-time

Build the future of the AI Data Cloud. Join the Snowflake team. We are looking for talented System Developers and Researchers to join the Snowflake AI Research team and contribute to LLM inference and training system development, optimizations, and agentic systems. Our mission is to build the most efficient and scalable generative AI systems. Recent releases from our team include SwiftKV, an advanced inference optimization, and Arctic LLM, one of the largest open-source MoE foundation models.

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

Senior Performance Engineer - Deep Learning

NVIDIA Corporation

Remote or Santa Clara, California, USA

Full-time

NVIDIA is hiring senior software engineers to build and optimize the tools Deep Learning engineers use across the world to design, develop, and deploy AI applications. We are an ambitious, forward-thinking and diverse team that influences all areas of NVIDIA's AI platform and directly contributes to premiere Deep Learning frameworks - PyTorch, JAX and TensorFlow. We work with multiple teams both inside and outside of NVIDIA across fields, as well as collaborate with the open-source community to

AI Research 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

Staff Software Engineer, Generative AI Platforms

Tesla Motors

Palo Alto, California, USA

Full-time

This Software Engineer will be at the forefront of developing & delivering cutting-edge applications on Tesla's advanced generative AI platform. Your primary focus will be on building & enhancing core components of the generative AI system, ensuring they are both efficient and scalable. You will collaborate closely with team members to translate complex research ideas into robust, production-ready services. This role requires a strong foundation in modern machine learning frameworks and a proven

Senior Software Engineer-ML Compiler

Ericsson Inc.

Remote or Austin, Texas, USA

Full-time

Grow with us Senior Software Engineer -- ML Compiler About this Opportunity This position is in Austin, Texas and is not a remote opportunity. Ericsson Inc. does not sponsor US work authorizations for this job position including H-1B, O-1, and TN. Ericsson also does not hire F-1's working onfor this position. Ericsson is seeking a software engineer specializing in ML compiler development to join the systems team. This role specializes on developing compiler tools for Ericsson silicon product

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

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

Deep Learning Engineer - Distributed Task-Based Backends

NVIDIA Corporation

Remote or Santa Clara, California, USA

Full-time

We are looking for Senior to Principal level experienced software professionals to help build the next generation of distributed backends for premier Deep Learning frameworks like PyTorch, JAX and TensorFlow. You will build on top of validated task-based runtime systems like Legate, Legion & Realm to develop a platform that can scale a wide range of model architectures to thousands of GPUs! What You Will Be Doing: Develop extensions to popular Deep Learning frameworks, that enable easy experime

Senior Software Engineer - Automated Parallel Programming

NVIDIA Corporation

Remote or Santa Clara, California, USA

Full-time

The PyTorch Team @ NVIDIA is hiring passionate parallel programmers. Join us to design and build the tools used by millions of AI practitioners deploying AI applications scalable to thousands of GPUs. Our team is responsible for the continual delivery of best in class experience on NVIDIA's hardware with PyTorch. Join our team and collaborate with many multi-disciplinary engineering teams within NVIDIA and internationally in the PyTorch open source community to deliver our customers the best of

Machine Learning Engineer (L5) - Content and Studio

Netflix, Inc.

Remote

Full-time

Netflix is one of the world's leading entertainment services, with 283 million paid memberships in over 190 countries enjoying TV series, films and games across a wide variety of genres and languages. Members can play, pause and resume watching as much as they want, anytime, anywhere, and can change their plans at any time. We are looking for an experienced ML engineer to develop, optimize, and deploy scalable ML solutions that enable self-service of analytics of both structured and unstructure

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

Manager, Research Science (Foundational Research, Machine Learning)

Thomson Reuters U.S. Inc.

Remote or Eagan, Minnesota, USA

Full-time

Are you a curious and open-minded individual with an interest in leading teams conducting state-of-the-art foundational machine learning research? Thomson Reuters Labs is seeking Research Scientist Managers with a passion for leading teams and solving challenging machine learning problems in a data-rich, complex academic environment driven by real-world problems. Foundational Research is the dedicated core Machine Learning research division of Thomson Reuters. We are focused on research and deve

Machine Learning Software Engineer L4/L5

Netflix, Inc.

Remote

Full-time

Netflix is one of the world's leading entertainment services, with 283 million paid memberships in over 190 countries enjoying TV series, films and games across a wide variety of genres and languages. Members can play, pause and resume watching as much as they want, anytime, anywhere, and can change their plans at any time. As Netflix continues to grow, so do the opportunities to enhance our personalization systems and algorithms. We're looking for a passionate and talented Software Engineer fo