JAX jobs in New York

Remote or New York, New York

•

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

Machine Learning Performance Engineer (Inference)

New York, New York

•

Today

Tower Research Capital is a leading quantitative trading firm founded in 1998. Tower has built its business on a high-performance platform and independent trading teams. We have a 25+ year track record of innovation and a reputation for discovering unique market opportunities. Tower is home to some of the world's best systematic trading and engineering talent. We empower portfolio managers to build their teams and strategies independently while providing the economies of scale that come from a

Full-time

USD 200,000.00 - 300,000.00 per year

Research Engineer, Pre-Training

New York, New York

•

Today

Jump Trading Group is committed to world-class research. We empower exceptional talents in Mathematics, Physics, and Computer Science to seek scientific boundaries, push through them, and apply cutting-edge research to global financial markets. Our culture is unique. Constant innovation requires fearlessness, creativity, intellectual honesty, and a relentless competitive streak. We believe in winning together and unlocking unique individual talent by incentivizing collaboration and mutual respec

Full-time

USD 300,000.00 - 350,000.00 per year

Machine Learning Engineer

New York, New York

•

10d ago

We are looking for an engineer with robust experience in machine learning and strong mathematical foundations to join our growing ML team and to help drive the direction of our ML platform. Machine learning is a critical pillar of Jane Street's global business. Our ever-evolving trading environment serves as a unique, rapid-feedback platform for ML experimentation, allowing us to incorporate new ideas with relatively little friction. Our ML team is full of people with a shared love for the craf

Full-time

AI Research Scientist - New York- Global Quant Firm

New York, New York

•

26d ago

Salary: up to 200k + bonus Summary Leading HFT fund looking for motivated research scientists with a demonstrated ability to apply machine learning to achieve cutting-edge capabilities in complex and challenging domains. You will join a growing AI team combining emerging techniques and models with original research to generate signals from unstructured data. In this role, you'll need to be capable of leading an open-ended research project from concept to production. This will include finding com

Full-time

USD 200,000.00 per year

MTS, Research Engineer

New York, New York

•

21d ago

About Us: Fireworks is the platform for specialized intelligence, enabling companies to build, train, and serve AI models tailored to their own data, workflows, and products. Founded by the team behind PyTorch and backed by AMD, Atreides, Benchmark Capital, Index Ventures, Lightspeed, NVIDIA, Sequoia Capital, and TCV, Fireworks powers production AI with hundreds of state-of-the-art open models across text, image, embedding, audio, and multimodal workloads. Today, Fireworks is a Series D company

Full-time

USD 250,000.00 - 400,000.00 per year

Senior Solutions Architect, Financial Services Capital Markets

New York, New York

•

Today

The Financial Services Solution Architect team is looking for an extraordinary person to join an experienced team of Quants and Data Scientists, engaging the finance industry with compelling examples of full-stack accelerated computing. Solution Architects work with top minds with Financial Services Capital Markets and Exchange firms to accelerate High-Performance Computing and AI workloads across various use cases. We're seeking an inquisitive, hard-working, and creative individual passionate a

Full-time

USD 224,000.00 - 356,500.00 per year

Senior Research Engineer - Voleon Securities

New York, New York

•

Today

Voleon Securities, a new business within the Voleon Group, provides liquidity in securities markets. We apply state-of-the-art AI/ML techniques to construct our liquidity-provision strategies. For nearly two decades, our affiliate Voleon Capital Management has led the hedge fund industry and worked at the frontier of applying AI/ML to investment management, becoming a multibillion-dollar asset manager. Voleon Securities builds on Voleon's deep real-world experience applying ML to financial marke

Full-time

USD 225,000.00 - 310,000.00 per year

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

Remote

•

Today

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

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

Easy Apply

Full-time

$225,000 - $275,000

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 is

Easy Apply

Full-time

$100,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% Remo

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 W

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 tec

Easy Apply

Full-time

$170,000 - $200,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 Ag

Easy Apply

Full-time

$130,000 - $175,000

Senior Research Engineer - Voleon Securities

New York, New York

•

10d ago

Full-time

Senior Manager, Technology Consulting

Remote

•

Today

Full-time

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

Lead Machine Learning Engineer, Entry and Re-engagement

New York, New York

•

Today

#WeAreParamount on a mission to unleash the power of content you in? We've got the brands, we've got the stars, we've got thepowerto achieve our mission to entertain the planet - now all we're missing is YOU! Becoming a part of Paramount means joining a team of passionate people who not only recognize the power of content but also enjoy a touch of fun and uniqueness. Together, we co-create moments that matter - both for our audiences and our employees - and aim to leave a positive mark on cultur

Full-time

USD 156,800.00 - 235,200.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

JAX jobs in New York, NY, USA