JAX Jobs

Refine Results
81 - 100 of 189 Jobs

Senior Machine Learning Engineer, Siri Attention and Invocation

Apple, Inc.

•

No location provided

Full-time

We are a group of engineers working on making Siri more intelligent and intuitive to interact with. We focus on innovating multimodal machine learning solutions that improve the experience for our users. We adapt our technologies based on various sensors available on different hardware products and make them run efficiently. We collaborate with various teams at Apple to deploy these innovative solutions to our products. Description As a member of the team you will be responsible for researching
Posted 60+ days ago | Updated 2 hours ago

Principal Generative-AI Software Engineer

NVIDIA Corporation

•

Santa Clara, California, USA

Full-time

At NVIDIA, we're not just building the future-we're generating it. Our Cosmos generative AI engineering team is pushing the boundaries of what's possible across multimodal learning, video generation, synthetic data, and intelligent simulation. We welcome hard-working engineers and applied scientists with deep experience in generative modeling to help define the next era of AI computing. What you'll be doing: Design, post-train, and optimize novel world models (e.g., diffusion video models, VLM
Posted 51 days ago | Updated 2 hours ago

Senior/Staff Software Engineer, ML Infrastructure, Optimization

Nuro Inc.

•

Mountain View, California, USA

Full-time

Who We Are Nuro exists to better everyday life through robotics. Founded in 2016, Nuro has spent eight years developing autonomous driving (AD) technology and commercializing AD applications. The Nuro Driver is our world-class autonomous driving system that combines AD hardware with our generalized AI-first self-driving software. Built to learn and improve through data, the Nuro Driver is one of the few driverless autonomous technologies on public roads today. Nuro has raised over $2B in capi
Posted 60+ days ago | Updated 2 hours ago

Principal 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 48 days ago | Updated 11 hours ago

AI Machine learning engineer

York Solutions, LLC

•

Chicago, Illinois, USA

Contract

Logistics: Type: 6+ month contract to hire (C2H is a must have) Location: Willis Tower, downtown Chicago, IL - 5 days on-site, after conversion only 4 days on-site required This team is looking for someone passionate about machine learning who can join a startup like division They are on the cutting edge of AI initiatives This team supports many ongoing AI initiative all centered around medical devices so healthcare/med device/pharma experience highly preferred. Role Summary Drive the technica
Easy Apply
Posted 21 hours ago

Senior AI Software Engineer, GenAI Framework

NVIDIA Corporation

•

Santa Clara, California, USA

Full-time

We are now looking for AI Software Engineers for NeMo! NVIDIA NeMo is an open-source, scalable and cloud-native framework built for researchers and developers working on Large Language Models (LLM), Multimodal (MM), and Speech AI. NeMo provides end-to-end model training, including data curation, alignment, customization, evaluation, deployment and tooling to optimize performance and user experience. In this critical role, you will expand NeMo Framework's capabilities, enabling users to develop,
Posted 45 days ago | Updated 2 hours ago

On-device ML Infrastructure Engineer (Core ML Framework)

Apple, Inc.

•

No location provided

Full-time

The On-Device Machine Learning team at Apple is responsible for enabling the Research to Production lifecycle of cutting edge machine learning models that power magical user experiences on Apple's hardware and software platforms. Apple is the best place to do on-device machine learning, and this team sits at the heart of that discipline, interfacing with research, SW engineering, HW engineering, and products. The team builds critical infrastructure that begins with onboarding the latest machine
Posted 60+ days ago | Updated 2 hours ago

Senior Deep Learning Software Engineer, Inference and Model Optimization

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

Machine Learning Engineer Intern (TikTok - P- LLM/Content Understanding) - 2025 Start (PhD)

TikTok

•

San Jose, California, USA

Full-time

Location : San Jose Employment Type : Intern Job Code : A45454 Apply to this job Share this listing: Responsibilities Our mission is to empower professional creators, media partners, and organizations to produce high-quality content that engages, entertains, and informs TikTok's diverse global audience. We are a product-facing team and you will work closely with a multi-disciplinary team (Product, Engineers, Design, and Data) to define and build solutions for creators who inspire members
Posted 60+ days ago | Updated 2 hours ago

Sr. Applied ML Engineer, Wallet & Payments

Apple, Inc.

•

No location provided

Full-time

Imagine what you can do here. Apple is a place where extraordinary people gather to do their lives' best work! In Wallet, Payments & Commerce, we create products and experiences people once couldn't have imagined, and now, can't imagine living without. It's the diversity of those people and their ideas that inspires the innovation that runs through everything we do. If you are a self-motivated, energetic individual with a track record of delivering impactful results, we want to hear from you! De
Posted 8 days ago | Updated 1 hour ago

Principal Engineer, AI/ML Software Compiler

Samsung Electronics America

•

San Jose, California, USA

Full-time

Please Note: To provide the best candidate experience amidst our high application volumes, each candidate is limited to 10 applications across all open jobs within a 6-month period. Advancing the World's Technology Together Our technology solutions power the tools you use every day--including smartphones, electric vehicles, hyperscale data centers, IoT devices, and so much more. Here, you'll have an opportunity to be part of a global leader whose innovative designs are pushing the boundaries o
Posted 56 days ago | Updated 2 hours ago

Lead Software Engineer - Machine Learning Engineer

JPMorgan Chase & Co.

•

Jersey City, New Jersey, USA

Full-time

Job Description Be an integral part of an agile team that's constantly pushing the envelope to enhance, build, and deliver top-notch technology products. Are you ready to make a significant impact in the world of AI and machine learning? At JPMorgan Chase, In this role, you'll be a key player in an agile team dedicated to enhancing, building, and delivering market-leading technology products that are secure, stable, and scalable As a Senior Lead Software/Machine Learning Engineer, at JPMorgan
Posted 30 days ago | Updated 2 hours ago

AIML - Senior Machine Learning Infrastructure Engineer -ML Compute, ML Platform & Technology

Apple, Inc.

•

No location provided

Full-time

Apple is where individual imaginations gather together, committing to the values that lead to great work. Every new product we build, service we create, or Apple Store experience we deliver is the result of us making each other's ideas stronger. That happens because every one of us shares a belief that we can make something wonderful and share it with the world, changing lives for the better. It's the diversity of our people and their thinking that inspires the innovation that runs through every
Posted 60+ days ago | Updated 2 hours ago

Senior Deep Learning Software Engineer, LLM Performance

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

Principal Deep Learning Software Engineer, LLM Performance

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

Java Full stack Developer (Integrated eligibility)

Apexon

•

Harrisburg, Pennsylvania, USA

Full-time

Skills: Java; J2EE); Oracle; REST Webservice, Integrated eligibility experience, Java (IDE). Key Responsibilities: Must have an integrated eligibility experience.Experience designing and working with n-tier architectures (UI, business logic layer, data access layer).Experience in developing REST and SOAP services using JAX-WS/RS and/or Spring framework.Experience in building multi module projects, compiling, building, and packaging the source code.Experience with TypeScript or EcmaScript is pre
Easy Apply
Posted 20 days ago | Updated 4 days ago

Software Engineer - Model Training Infrastructure - USDS

TikTok

•

Seattle, Washington, USA

Full-time

Location : Seattle Employment Type : Regular Job Code : A44296 Apply to this job Share this listing: Responsibilities About the team The mission of our AML team is to push the next-generation AI infrastructure and recommendation platform for the ads ranking, search ranking, live & ecom ranking in our company. We also drive substantial impact on core businesses of the company. Currently, we are looking for Machine Learning Engineer - Model Training Infrastructure to join our team to suppor
Posted 59 days ago | Updated 2 hours ago

Senior Director - AI Solutions and Emerging Technology

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

Model Optimization Engineer (Algorithmic development)

Apple, Inc.

•

No location provided

Full-time

Are you excited about the impact that optimizing deep learning models can have on enabling transformative user experiences? The field of ML compression research continues to grow rapidly and new techniques to perform quantization, pruning etc are increasingly available to be ported and adopted by the ML developer community looking to ship more models in a constrained memory budget and make them run faster. We are passionate about productizing and pushing the envelope of the state of the art mode
Posted 60+ days ago | Updated 2 hours ago

Senior DGX Cloud AI Infrastructure Software Engineer

NVIDIA Corporation

•

Remote or Santa Clara, California, USA

Full-time

Joining NVIDIA's DGX Cloud AI Efficiency Team means contributing to the infrastructure that powers our innovative AI research. This team focuses on optimizing efficiency and resiliency of AI workloads, as well as developing scalable AI and Data infrastructure tools and services. Our objective is to deliver a stable, scalable environment for AI researchers, providing them with the necessary resources and scale to foster innovation. We are seeking an AI infrastructure software engineer to join our
Posted 51 days ago | Updated 2 hours ago