JAX jobs in San Francisco, CA

Staff Software Engineer, ML Infra, Autonomy

Palo Alto, California

•

Today

About Rivian Rivian is on a mission to keep the world adventurous forever. This goes for the emissions-free Electric Adventure Vehicles we build, and the curious, courageous souls we seek to attract. As a company, we constantly challenge what's possible, never simply accepting what has always been done. We reframe old problems, seek new solutions and operate comfortably in areas that are unknown. Our backgrounds are diverse, but our team shares a love of the outdoors and a desire to protect it

Full-time

USD 206,500.00 - 258,100.00 per year

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

Remote

•

Today

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

Easy Apply

Full-time

$160,000 - $190,000

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

Remote

•

Today

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

Easy Apply

Full-time

$150,000 - $170,000

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

Remote

•

Today

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

Easy Apply

Full-time

$190,000 - $225,000

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

Remote

•

Today

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 te

Easy Apply

Full-time

$170,000 - $200,000

LLM Application Engineer

San Francisco, California

•

Today

Software Guidance & Assistance, Inc., (SGA), is searching for an LLM Application Engineer for a Remote Full Time assignment with one of our premier Technology clients. Responsibilities : Design, develop, and deploy applications powered by large language models and intelligent workflow automation. Create agent-based systems that support reasoning, planning, memory management, tool integration, and multi-step task execution. Build orchestration frameworks that transform AI-generated outputs into d

Full-time

USD 120,000.00 - 205,000.00 per year

Principal AI Infrastructure & Accelerator Architect

Remote or New York, New York

•

Today

Are you a visionary technical leader passionate about defining the future of AI infrastructure and squeezing every drop of performance out of advanced hardware accelerators at scale? We are seeking a Principal Architect to lead, shape, and execute our technical strategy for AI performance, optimization, and hardware-software co-design. In this elite, highly visible role, you will define the architectural vision for both AI training and serving infrastructure, delivering massive industry-wide imp

Full-time

Remote Machine Learning Researcher

Remote

•

Today

Job Description A quantitative investment and research organization is seeking a Machine Learning Researcher / Engineer to join a highly collaborative, hands-on team. This is a fully remote opportunity with compensation up to $200,000 based on experience. This role is ideal for someone with a strong mathematical foundation who enjoys evaluating different machine learning approaches, transforming research into production-quality code, and delivering practical results. The team values intellectua

Easy Apply

Full-time

USD 200,000.00 per year

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

Remote

•

Today

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) Responsibilities:

Easy Apply

Full-time

$225,000 - $275,000

Machine Learning Lead Engineer, Machine Learning (ML) Required, Work From Home - G

Remote

•

Today

Machine Learning Lead Engineer, Machine Learning (ML) Required, Work From Home As Machine Learning Lead Engineer, you own the execution layer of our intelligence, turning research and model capabilities into reliable, scalable production systems. You will work across the model lifecycle: data, training, evaluation, inference, and deployment. This is a hands-on leadership role for someone who wants to operate at the intersection of research, systems, and product. Machine Learning (ML) experien

Easy Apply

Full-time

$160,000 - $190,000

Machine Learning (ML) Engineer, Ranking & Retrieval, Machine Learning (ML) and Artificial Intelligence (AI) Required, Work From Home - M

Remote

•

Today

Machine Learning (ML) Engineer, Ranking & Retrieval, Machine Learning (ML) and Artificial Intelligence (AI) Required, Work From Home The Machine Learning (ML) Engineer will build and improve Machine Learning (ML) systems that power our AI-native applications. You will work on classification, retrieval, ranking, personalization and evaluation, taking ownership from model development to production deployment. This role is for engineers who enjoy solving practical Machine Learning (ML) problems,

Easy Apply

Full-time

$120,000 - $160,000

Applied Machine Learning (ML) Engineer, Machine Learning (ML) and Artificial Intelligence (AI) Required, Work From Home - M

Remote

•

Today

Applied Machine Learning (ML) Engineer, Machine Learning (ML) and Artificial Intelligence (AI) Required, Work From Home The Applied Machine Learning (ML) Engineer will build and improve Machine Learning (ML) systems that power our AI-native applications. You will work on classification, retrieval, ranking, personalization and evaluation, taking ownership from model development to production deployment. This role is for engineers who enjoy solving practical Machine Learning (ML) problems, impro

Easy Apply

Full-time

$120,000 - $160,000

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

Remote

•

Today

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 A

Easy Apply

Full-time

$130,000 - $175,000

Machine Learning Platform Engineer, Machine Learning (ML) and Artificial Intelligence (AI) Required, Work From Home - G

Remote

•

Today

Machine Learning Platform Engineer, Machine Learning (ML) and Artificial Intelligence (AI) Required, Work From Home As the Machine Learning Platform Engineer, you will build the infrastructure and systems that power Artificial Intelligence (AI) capabilities. You will design and operate the systems behind the Artificial Intelligence (AI) stack, from model training and evaluation to deployment, inference, observability, and continuous improvement. You will work closely with Artificial Intelligen

Easy Apply

Full-time

$140,000 - $180,000

Staff Software Engineer, Machine Learning (Consumer Revenue)

Remote or San Francisco, California

•

Today

Discord has a highly engaged community of millions of daily active users who use the platform for many different reasons, but there's one thing that nearly everyone does: play video games. Discord plays a uniquely important role in the future of gaming, and we are focused on making it easier and more fun for people to hang out before, during, and after playing games. We're looking to hire a Staff Machine Learning Engineer on our Consumer Revenue ML team. This team applies Machine Learning acros

Full-time

USD 272,000.00 - 340,000.00 per year

Senior Software Engineer, Machine Learning (Safety)

San Francisco, California

•

Today

Discord has a highly engaged community of millions of daily active users who use the platform for many different reasons, but there's one thing that nearly everyone does: play video games. Discord plays a uniquely important role in the future of gaming, and we are focused on making it easier and more fun for people to hang out before, during, and after playing games. We are seeking an experienced Senior Machine Learning Engineer to join our Safety ML team. This role focuses on building and depl

Full-time

USD 220,000.00 - 275,000.00 per year

PhD Machine Learning Software Engineer Intern (Summer 2027)

San Francisco, California

•

Today

At Lyft, our purpose is to serve and connect. We aim to achieve this by cultivating a work environment where all team members belong and have the opportunity to thrive. With over half a billion rides and counting, Lyft is solving hard problems in a flourishing domain with a lot of data and creative solutions in Marketplace, Mapping, Fraud, Growth and beyond. We're actively building the next-generation Machine Learning (ML) platform for low-cost, ultra-immersive transportation to improve people'

Full-time

USD 65.00 - 68.00 per hour

Machine Learning (ML) Engineer, Ranking & Retrieval, Machine Learning (ML) and Artificial Intelligence (AI) Required, Work From Home

Remote or San Francisco, California

•

Today

Company Description Job Description Machine Learning (ML) Engineer, Ranking & Retrieval, Machine Learning (ML) and Artificial Intelligence (AI) Required, Work From Home The Machine Learning (ML) Engineer will build and improve Machine Learning (ML) systems that power our AI-native applications. You will work on classification, retrieval, ranking, personalization and evaluation, taking ownership from model development to production deployment. This role is for engineers who enjoy solving pract

Full-time

Applied Machine Learning (ML) Engineer, Machine Learning (ML) and Artificial Intelligence (AI) Required, Work From Home

Remote or San Francisco, California

•

Today

Company Description Job Description Applied Machine Learning (ML) Engineer, Machine Learning (ML) and Artificial Intelligence (AI) Required, Work From Home The Applied Machine Learning (ML) Engineer will build and improve Machine Learning (ML) systems that power our AI-native applications. You will work on classification, retrieval, ranking, personalization and evaluation, taking ownership from model development to production deployment. This role is for engineers who enjoy solving practical

Full-time

ML Ops Engineer

South San Francisco, California

•

Today

At Lilly, the work is demanding because patients are waiting. We unite caring with discovery to help make life better for people around the world, knowing that every decision, every detail, and every day matters. Headquartered in Indianapolis, Indiana, our over 50,000 employees around the globe take on complex challenges to discover and deliver life-changing medicines, strengthen how health is understood and managed, and support the communities we serve. This is hard, urgent, selfless work-but i

Full-time

USD 147,000.00 - 268,400.00 per year

JAX jobs in San Francisco, CA, USA