jax Jobs in california

Refine Results
21 - 39 of 39 Jobs

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

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

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

Java Software Engineer

SAIC

Remote

Full-time

Job ID: 2504034 Location: REMOTE WORK, VA, US Date Posted: 2025-04-29 Category: Software Subcategory: SW Engineer Schedule: Full-time Shift: Day Job Travel: No Minimum Clearance Required: None Clearance Level Must Be Able to Obtain: Public Trust Potential for Remote Work: Yes Description Job Description: SAIC is seeking an experienced Software Engineer to join our dynamic Agile team delivering software solutions to the United States Patent and Trademark Office. In this role you will d

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

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

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

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

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

AI and ML Infra Software Engineer, GPU Clusters

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.

Software Engineer, Computer Vision

SpaceX

Hawthorne, California, USA

Full-time

SpaceX was founded under the belief that a future where humanity is out exploring the stars is fundamentally more exciting than one where we are not. Today SpaceX is actively developing the technologies to make this possible, with the ultimate goal of enabling human life on Mars. SOFTWARE ENGINEER, COMPUTER VISION We are looking for good people and driven, adaptable, resilient software engineers with experience developing software to accelerate quality, speed, and efficiency of manufacturing p

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

Software Engineering Manager, Sparse Linear Algebra Libraries

NVIDIA Corporation

Remote or Santa Clara, California, USA

Full-time

We are looking for a software engineering manager to lead our development efforts for high-performance sparse linear algebra libraries in the AI and scientific computing domains. Around the world, leading commercial and academic organizations are revolutionizing AI, data analytics, and scientific and engineering simulations, using data centers powered by GPUs. Applications of these technologies include LLMs, computer vision, encryption, computer aided engineering (CAE), electronic design automat

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