JAX jobs in Texas

Remote

•

Today

Are you passionate about squeezing every last drop of performance out of advanced hardware accelerators? Step into a role where you will shape the future of AI. In this role, you will drive the performance and optimization of both training and serving, delivering massive impact for customers. We are seeking a talented leader to build out a center of excellence and scale this practice. You will have exposure to the newest Tensor Processing Unit (TPU) and Graphics Processing Unit (GPU) hardware, t

Full-time

Senior Manager, Technology Consulting

Remote or New York, New York

•

Today

Full-time

Python Engineer - Assistant Vice President

Irving, Texas

•

Today

Citi, the leading global bank, has approximately 200 million customer accounts and does business in more than 160 countries andjurisdictions. 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. As a bank with a brain and a soul, Citi createseconomicvalue that is systemically responsible

Full-time

Compensation information provided in the description

VP of Research, Machine Learning (ML), Artificial Intelligence (AI) Required, Work From Home - M

Remote

•

2d ago

VP of Research, Machine Learning (ML), Artificial Intelligence (AI) Required, Work From Home The VP of Research, Machine Learning (ML) will own the research and intelligence direction of this system. The VP of Research, Machine Learning (ML) will define how Artificial Intelligence (AI) reasons, evaluates, and improves in a product used with high frequency. This position is 100% Remote. MUST BE WILLING TO TAKE A 60 MINUTE CODING ASSESSMENT. VP of Research, Machine Learning (ML) Responsibilitie

Easy Apply

Full-time

$225,000 - $275,000

Machine Learning Technical Lead, Artificial Intelligence (AI) Required, Work From Home - G

Remote

•

2d ago

Machine Learning Technical Lead, Artificial Intelligence (AI) Required, Work From Home As Machine Learning Technical Lead, you own the execution layer of intelligence. You will translate research direction into reliable, scalable, production-grade ML systems. This role sits at the intersection of research, infrastructure, and product. You will be responsible for making models trainable, deployable, observable, and performant under real-world constraints. This position is 100% Remote. MUST BE W

Easy Apply

Full-time

$190,000 - $225,000

LLM Application Engineer, Artificial Intelligence (AI) Required, Work From Home - M

Remote

•

2d ago

LLM Application Engineer, Artificial Intelligence (AI) Required, Work From Home As an LLM Application Engineer, you will build the intelligence layer that powers the AI experiences. You will work at the intersection of Large Language Models (LLMs), software engineering, and product designing agent workflows, improving model behavior, and turning Artificial Intelligence (AI) capabilities into reliable user experiences. You will own problems end-to-end, from understanding user needs, designing Ag

Easy Apply

Full-time

$130,000 - $175,000

Principal Machine Learning Engineer, Artificial Intelligence (AI) Required, Work From Home - G

Remote

•

2d ago

Principal Machine Learning Engineer, Artificial Intelligence (AI) Required, Work From Home As a Principal Machine Learning Engineer, you are a deep technical authority responsible for designing and evolving the most critical ML systems in the company. The Principal Machine Learning Engineer will operate across training, inference, evaluation, and infrastructure, solving the hardest architectural and performance problems. While Technical Leads may own execution at the team level, you set the tec

Easy Apply

Full-time

$170,000 - $200,000

Member of Technical Staff, Machine Learning, Artificial Intelligence (AI) Required, Work From Home - G

Remote

•

2d ago

Member of Technical Staff, Machine Learning, Artificial Intelligence (AI) Required, Work From Home As a Member of Technical Staff, Machine Learning, you will build core ML components. The Member of Technical Staff will work on real production systems from day one, learning how large-scale ML behaves outside of research settings. The Member of Technical Staff role is for engineers who want to develop strong systems judgment by shipping, debugging, and iterating on real-world ML. This position is

Easy Apply

Full-time

$100,000 - $190,000

Staff Machine Learning Engineer, Artificial Intelligence (AI) Required, Work From Home - G

Remote

•

2d ago

Staff Machine Learning Engineer, Artificial Intelligence (AI) Required, Work From Home As the Staff Machine Learning Engineer, you own the execution layer of intelligence. You translate research direction into reliable, scalable, production-grade Machine Learning (ML) systems. This role sits at the intersection of research, infrastructure, and product. You are responsible for making models trainable, deployable, observable, and performant under real-world constraints. This position is 100% Remo

Easy Apply

Full-time

$150,000 - $170,000

Senior Machine Learning Engineer, Artificial Intelligence (AI) Required, Work From Home - G

Remote

•

2d ago

Senior Machine Learning Engineer, Artificial Intelligence (AI) Required, Work From Home As the Senior Machine Learning Engineer, you are an independent owner of critical Machine Learning (ML) subsystems in production. You take ambiguous problems, design practical solutions, and ship systems that operate reliably at scale. This is a hands-on, high-impact role focused on depth. This position is 100% Remote. MUST BE WILLING TO TAKE A 60 MINUTE CODING ASSESSMENT. Senior Machine Learning Engineer

Easy Apply

Full-time

$170,000 - $190,000

Lead Data Scientist - Autonomous Goal Management

Remote

•

Today

Become a part of our caring community The Enterprise AI organization at Humana is a pioneering force, driving AI innovation across our Insurance and CenterWell business segments. By collaborating with world-leading experts, we are at the forefront of delivering cutting-edge AI technologies for improving care quality and experience of millions of consumers. We are actively seeking top talent to develop robust and reusable AI modules and pipelines, ensuring adherence to best practices in accountab

Full-time

USD 142,300.00 - 195,700.00 per year

Machine Learning Engineer, Data Mining

Remote or Boston, Massachusetts

•

Today

Mission Summary: At Motional, we're transforming how autonomous vehicles discover critical intelligence hidden within petabytes of multimodal sensor data. Our next-generation autonomous driving stack depends on finding the rare edge cases, long-tail scenarios, and model errors that matter most. Omnitag, our ML-powered multimodal data mining framework, is the engine that powers this discovery. As a Machine Learning Engineer on the Data Mining team, your mission is to help build the "Brain" of t

Full-time

USD 144,000.00 - 192,000.00 per year

ML Features Solutions Engineer

Austin, Texas

•

19d ago

The era of pervasive AI has arrived. In this era, organizations will use generative AI to unlock hidden value in their data, accelerate processes, reduce costs, drive efficiency and innovation to fundamentally transform their businesses and operations at scale. SambaNova Suite is the first full-stack, generative AI platform, from chip to model, optimized for enterprise and government organizations. Powered by the intelligent SN40L chip, the SambaNova Suite is a fully integrated platform, deliv

Full-time

USD 200,000.00 - 270,000.00 per year

Staff Data Scientist | NLP

Remote

•

Today

Machinify is a leading healthcare intelligence company with expertise across the payment continuum, delivering unmatched value, transparency, and efficiency to health plan clients across the country. Deployed by over 85 health plans, including many of the top 20, and representing more than 270 million lives, Machinify brings together a fully configurable and content-rich, AI-powered platform along with best-in-class expertise. We're constantly reimagining what's possible in our industry, creatin

Full-time

USD 180,000.00 - 230,000.00 per year

Senior ML Engineer

Remote

•

Today

We are looking for a Machine Learning Engineer with practical engineering skills and a deep understanding of modern AI systems, including LLMs, RAG architectures, agents, and safety considerations.Success in this role requires the ability to work in ambiguous environments, define measurable objectives, create evaluation methodologies when none exist, and rapidly iterate toward effective solutions. Essential functions Own machine learning projects from problem definition through implementation.D

Full-time

Reinforcement Learning Engineer, Grasping

Houston, Texas

•

Today

Persona AI is developing and commercializing rugged, multi-purpose humanoid robots that perform real work. Persona's founding team has a decades-long history in humanoid robotics, bionics, and product development delivering robust hardware that has touched the stars, worked miles below the surface of the ocean, roamed Disney Parks, and has even been featured on a US postage stamp. Our mission is focused squarely on shipping beautiful, reliable products at massive scale, while building a customer

Full-time

FWI & AI Scientist

Houston, Texas

•

Today

Job Title: FWI & AI Scientist Lead the development of the FWI Foundation Model Design and scale a generalizable hybrid physics-AI foundation model on real seismic datasets. Propose advanced techniques including AI-assisted velocity model building, learned regularization and preconditioning, cycle-skipping mitigation, and accelerated forward modeling/gradient computation. Scale implementations for GPU clusters, HPC systems, and large-scale 3D datasets.Integrate AI into FWI and imaging pipelines

Full-time

Senior Software Engineer - AI Middleware

Austin, Texas

•

Today

At Cornelis we're building the future of AI and HPC networking with an AI-first approach to silicon and software development. We're seeking engineers who are energized by working on cutting-edge ASIC design and distributed software systems, and who are motivated to push the boundaries on how AI can transform everything from chip architecture to system performance at scale. Cornelis Networks delivers the world's highest performance scale-out networking solutions for AI and HPC datacenters. Our d

Full-time

Python Engineer - Assistant Vice President

Irving, Texas

•

Today

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. As a bank with a brain and a soul, Citi creates economic value that is systemically responsib

Full-time

USD 107,120.00 - 160,680.00 per year

Principal System Software Architect, AI/GPU Platforms

Austin, Texas

•

Today

WHAT YOU DO AT AMD CHANGES EVERYTHING At AMD, our mission is to build great products that accelerate next-generation computing experiences-from AI and data centers, to PCs, gaming and embedded systems. Grounded in a culture of innovation and collaboration, we believe real progress comes from bold ideas, human ingenuity and a shared passion to create something extraordinary. When you join AMD, you'll discover the real differentiator is our culture. We push the limits of innovation to solve the wo

Full-time

JAX jobs in Texas, USA