jax jobs in new york

Member of Technical Staff (AI Inference Engineer)

New York, New York

•

Today

We build and run the inference engine behind every Perplexity query and deploy dozens of model architectures at scale with tight latency and cost budgets. Our stack is Rust, Python, CUDA, and CuTe DSL - and we need another engineer to join us. What you will work on Examples of real work the team does: New models support. Support transformer-based retrieval, text-generation, and multimodal models in our inference infrastructure, from weight loading, request scheduling and KV-cache management to

Full-time

USD 220,000.00 - 485,000.00 per year

Data Scientist, Algorithms - Lyft Ads

New York, New York

•

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. Lyft Ads is one of Lyft's newest and fastest-growing businesses, focused on building the world's largest transportation media network. Our mission is to help brands reach riders during key moments of their journey-before, during, and after a ride-by delivering meaningful, contextually relevant ad experiences. We operate at the in

Full-time

USD 128,000.00 - 160,000.00 per year

Machine Learning Engineer

New York, New York

•

6d 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 craft

Full-time

Senior Machine Learning Engineer

New York, New York

•

7d ago

About the Role A1 is building a proactive AI chat app for everyday users to bring intelligence to conversations, errands, organising and workflows. Unlike traditional chat-based applications, our product focuses on achieving high reliability for long-running workflows, persistent context, and real-world task completion. The system must handle multi-step reasoning, interact with external tools, and remain reliable despite non-deterministic model behavior. As a Senior Member of Technical Staff, Ma

Full-time

Machine Learning Engineer

New York, New York

•

Today

Full-time

Staff Machine Learning Engineer

New York, New York

•

7d ago

Full-time

AI Engineer / Data Scientist, AI Manager

New York, New York

•

Today

Industry/Sector Not Applicable Specialism Assurance Management Level Manager Job Description & Summary At PwC, our people in audit and assurance focus on providing independent and objective assessments of financial statements, internal controls, and other assurable information enhancing the credibility and reliability of this information with a variety of stakeholders. They evaluate compliance with regulations including assessing governance and risk management processes and related controls.

Full-time

USD 99,000.00 per year

AI Research Engineer - Datadog AI Research (DAIR)

New York, New York

•

Today

As a Research Engineer on our team, you will partner with Research Scientists to turn research ideas into working systems, building the data, tooling, and infrastructure that enable rapid iteration, trustworthy evaluation, and a smooth path from prototype to production. Building on our track record of AI-powered solutions (e.g., Bits AI, Bits Evolve, and our time series foundation model), Datadog AI Research tackles high-risk, high-reward problems grounded in real-world challenges in cloud obse

Full-time

USD 140,000.00 per year

Java Mainframe Developer

Remote

•

12d ago

Job Title: Java Developer Location: Remote Type: Contract Responsibilities Design, develop, and enhance complex software applications supporting mission-critical, time-sensitive initiatives focused on integrating distributed applications with the organization''s Core Banking platform.Translate business and platform requirements into secure, scalable, and reliable solutions throughout the full software development lifecycle.Operate with minimal supervision while exercising strong technical judg

Easy Apply

Contract, Third Party

Depends on Experience

Senior Machine Learning Engineer, Dash Agentic AI

Remote

•

4d ago

Role Description As a Senior Machine Learning Engineer, you will play a key role in advancing Dropbox's mission to create a more enlightened way of working. Leveraging cutting-edge AI/ML technologies, you will design, build, deploy, and refine highly reliable AI agents operating at massive scale. Your work will power Dropbox Dash's universal agentic search and autonomous organization features, transforming how millions of users collaborate, stay organized, and focus on the work that truly matte

Full-time

USD 245,200.00 - 331,800.00 per year

Senior ML Compiler Engineer

Remote or Sunnyvale, California

•

Today

Job Description About the Mission GM's vision of Zero Crashes, Zero Emissions, and Zero Congestion guides everything we do in autonomous and assisted driving. The AV organization is building advanced automated driving technologies, including Level 4-capable fully self-driving systems, to move us toward safer, more sustainable, and more accessible mobility. For the AI Kernels & Compilers team, that mission shows up in the details: turning cutting-edge perception, prediction, and planning researc

Full-time

USD 128,700.00 - 261,300.00 per year

Machine Learning Engineer, Motion Planning & Prediction

Remote or Austin, Texas

•

Today

About the Team Our team develops the core software and data processing systems that power motion planning and decision-making in autonomous vehicles. We work at the intersection of machine learning, large-scale data infrastructure, and real-time vehicle control, collaborating across engineering, analytics, and product teams to deliver safe and intelligent driving capabilities. About the Role We are looking for a creative & driven Machine Learning Engineer to join our autonomous vehicle team. Y

Full-time

Staff ML Compiler Engineer

Remote or Sunnyvale, California

•

Today

Full-time

USD 185,100.00 - 335,300.00 per year

ML Validation Engineer - Early Career

Remote or Sunnyvale, California

•

Today

Job Description The ML Validation Research Engineer - Early Career will contribute to applied machine learning research focused on improving verification and validation of ML components used in robotics and autonomous driving systems. This role centers on simulation-based evaluation, performance monitoring and issue observability, uncertainty modeling, scenario coverage automation, and transforming advanced ML research into working prototypes that enhance the efficiency, accuracy, and coverage

Full-time

Sr. Artificial Intelligence and Machine Learning Scientist, Early Career

Remote or Mountain View, California

•

Today

Job Description Work Arrangement : This role is categorized as hybrid. This means the successful candidate is expected to report to the office three times per week or any other frequency dictated by the business. Location: Mt. View, California - Mountain View Technical Center About the Team: Our AI Research team, reporting directly to the Chief AI Officer, is pioneering how cutting-edge machine learning can transform the way vehicles are designed, manufactured, and experienced. We are building

Full-time

USD 140,000.00 - 195,000.00 per year

Senior ML Engineer - Embodied AI Onboard Autonomy

Remote or Mountain View, California

•

Today

Job Description At General Motors, our product teams are redefining mobility. Through a human-centered design process, we create vehicles and experiences that are designed not just to be seen, but to be felt. We're turning today's impossible into tomorrow's standard -from breakthrough hardware and battery systems to intuitive design, intelligent software, and next-generation safety and entertainment features. Every day, our products move millions of people as we aim to make driving safer, smart

Full-time

USD 158,000.00 - 241,900.00 per year

Senior AI Compiler Engineer

Remote or Austin, Texas

•

Today

Grow with us Senior AI Compiler Engineer - Research Team Austin, Texas This is not a remote work opportunity. MLIR Architecture | Compiler Infrastructure | 5G/6G Silicon | Bare-Metal AI The Mission Most AI compiler engineers optimize models for GPUs that already exist. You're going to optimize them for silicon that doesn't yet. As our Senior AI Compiler Engineer, you're not porting a framework, tweaking a runtime, or wrapping CUDA libraries. You're doing something far rarer and far harder -

Full-time

Staff ML Engineer - Embodied AI Onboard Autonomy

Remote or Mountain View, California

•

Today

Job Description Are you passionate about accelerating the future of autonomous driving? Join the Embodied AI team at General Motors. Our team is developing and deploying machine learning solutions that support safe and reliable autonomous vehicle behavior across real-world scenarios. As a Staff AI/ML Engineer 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 p

Full-time

USD 180,000.00 - 280,000.00 per year

Senior ML Validation Engineer

Remote or Sunnyvale, California

•

Today

Job Description The Senior ML Validation Research Engineer will lead applied machine learning research focused on improving verification and validation of ML components used in robotics and autonomous driving systems. This role centers on simulation-based evaluation, uncertainty modeling, scenario coverage automation, and transforming advanced ML research into working prototypes that enhance the efficiency, accuracy, and coverage of ML system validation. Key Responsibilities Prototype research

Full-time

USD 144,700.00 - 261,300.00 per year

Machine Learning Engineer

Remote or Austin, Texas

•

Today

About the Team Avride develops autonomous vehicle and delivery robot technology, leveraging deep expertise in autonomous systems. With the recent launch of our robotaxi service in Dallas, we are accelerating innovation and redefining the future of mobility. Our team builds self-driving solutions from the ground up, with machine learning at the core of our development pipeline to enable safe and intelligent navigation. We design and deploy state-of-the-art models to address key challenges in au

Full-time

Filter Results

Job post features

Posted date

Work settings

Employment type

Distance

Employer type

Work authorization

jax jobs in New York, NY, USA