Find work in the United States.

A simpler way to find your next job.

Jobs in the United States

25+ jobs

Save this search

Added 1 hour ago
Mechanical Engineer
Smith Johnson Tech
West Jordan, UT
Full-Time Mechanical Engineer position in our clients’ West Jordan, UT, office. Opportunity with an established, market leading, employee-owned company doing major innovative design…
Added 1 hour ago
AI Research Engineer (Agentic Reinforcement Learning)
Jobgether
Remote, United States
…applications and next steps. Our partner is looking for an AI Research Engineer (Agentic Reinforcement Learning) based in United…
Added 1 hour ago
PD Methodology Engineer, Annapurna Labs - Cloud Scale Machine Learning
Annapurna Labs (U.S.) Inc.
Austin, TX · Cupertino, CA
Custom SoCs (System on Chip) live at the heart of Amazon Machine Learning servers. As a member of the Cloud-Scale Machine Learning Acceleration team you’ll be responsible for the design and…
Added 1 hour ago
Senior Engineer, SoC HAL
Annapurna Labs (U.S.) Inc.
Austin, TX · Cupertino, CA
Annapurna Labs builds the custom silicon at the heart of AWS machine learning servers, and we are hiring a senior engineer to own the architecture of the Hardware Abstraction Layer (HAL)…
Added 1 hour ago
Forward Deployment (FD) Agentic Solutions Consultant - Sales
Jobgether
Remote, United States
Design and build prototypes: Rapidly develop functional agentic AI prototypes using real AI tools, synthetic data, and simulated systems that reflect clients' operational environments.
Added 1 hour ago
Computer Vision Engineer II
California Institute Of Technology Jet Propulsion Laboratory
United States
Wanted: Computer Vision (CV) Engineer to join the Navigation and Mapping group in the Guidance and Control Section.
Added 2 hours ago
Staff Engineer, Machine Learning & Data Infrastructure
OKX
San Jose, CA
We are looking for an experienced engineer to lead the architecture and hands-on development of the ML and data infrastructure behind our risk and fraud prevention systems. You will help…
Added 2 hours ago
Industrial Manufacturing Engineer
North American Stamping Group
Portland, TN
The Industrial Manufacturing Engineer reports to the Manufacturing Engineering Manager. Industrial Manufacturing Engineer is responsible to assist with new manufacturing processes and…
Added 2 hours ago
J
Enterprise Solution Architect
Jobgether
Remote, United States
- $210,000 to $250,000 USD / year
Define enterprise architecture principles, technical standards, governance frameworks, reference architectures, reusable design patterns, and Architecture Decision Records.
Added 3 hours ago
Manufacturing Engineer I
Panasonic
Sparks, NV
Do you want to join a team that's changing the world? Do you have a strong background as a Manufacturing Engineer I? Then we're looking for you! Check out the job description and apply now!…
Added 3 hours ago
Lead Specialist, AI Scientist
Pearson
Remote, United States
We are seeking a Lead Specialist, Applied AI Systems Engineer, to design, build, and continuously improve the intelligent systems that support Pearson’s work developing the next generation…
Added 3 hours ago
M
Senior Software Engineer
Microsoft
United States
- $119,800 to $261,000 USD / year
As a Senior Software Engineer, you will lead the design and development of a new solution that orchestrates Virtual Machine Scale Sets across Azure regions. The solution will help…
Added 3 hours ago
M
Software Engineer- CoreAI
Microsoft
WA · Redmond, WA · Redmond
- $85,400 to $183,700 USD / year
As a Software Engineer in CoreAI, you will collaborate with engineers, applied scientists, and product teams to build AI-powered developer experiences, intelligent applications, and…
Added 3 hours ago
M
Senior Machine Learning Engineer - Applied Scientist
Microsoft
WA · CA · NY · Redmond, WA · Mountain View, CA · San Francisco, CA · Redmond · Mountain View · New York · San Francisco
- $119,800 to $261,000 USD / year
We are looking for a Senior Machine Learning Engineer - Applied Scientist to help define, evaluate, and build the next generation of AI-powered product experiences. This role requires a…
Added 3 hours ago
M
Senior Software Engineer - Minecraft
Microsoft
WA · Redmond, WA · Redmond
- $119,800 to $261,000 USD / year
The Mojang Player Operations team is looking for a Senior Software Engineer to help drive innovation in Trust and Safety. Our team is accountable for ensuring our players have a world-class…
Added 3 hours ago
M
Senior Researcher/Engineer: AI for Biomolecular Modelling and Simulation
Microsoft
United States
- $119,800 to $261,000 USD / year
We invite you to join us on this mission. We are seeking Senior Researcher/Engineer with backgrounds in Machine Learning, Computational Chemistry, Physics or Biology. You will be working in…
Added 3 hours ago
M
Software Engineer 2
Microsoft
WA · CA · NY · Redmond, WA · Mountain View, CA · Redmond · Mountain View · New York
- $102,100 to $219,200 USD / year
…video, and multimodal content. We are looking for a Software Engineer 2 - Responsible AI who is passionate about building customer-facing AI services with scalable and sustainable…
Added 3 hours ago
Senior Machine Learning (ML) Engineer (CCC Casualty / Claims Intelligence)
EvolutionIQ
Remote, United States
As a Senior Machine Learning Engineer, you'll own models and evaluations end to end within a problem area and turn the results into product decisions. You'll join a new engineering group at…
Added 3 hours ago
PN
Research Engineer II - Building Energy Analysis
Pacific Northwest National Laboratory
UNAVAILABLE · Richland, WA · Seattle, WA · Portland, OR
- $89,300 USD / year
This position is in the Building Simulation and Design Group (BS&DG) within PNNL’s Electricity Infrastructure and Buildings Division. BS&DG conducts modeling and analysis to evaluate the…
Added 4 hours ago
Senior AI Engineer – Claude & Databricks
Techstra Solutions
Pittsburgh, PA
Design and develop data pipelines supporting AI, analytics, and machine learning workloads.
Added 4 hours ago
Mid-Level Databricks Engineer
Techstra Solutions
Pittsburgh, PA
We are seeking a Mid-Level Databricks Engineer to design, develop, and support scalable data solutions using Databricks and modern cloud data platforms. This role will focus on building…
Added 4 hours ago
Senior Databricks Engineer
Techstra Solutions
Pittsburgh, PA
We are seeking a Senior Databricks Engineer / Technical Lead with 10+ years of experience in data engineering, data warehousing, and ETL/ELT, including significant hands-on experience…
Added 4 hours ago
Machine Learning Engineer Senior
Techstra Solutions
Dallas, TX
Leads in the development of cross-functional, multi-platform machine learning software systems. Uses technical knowledge and industry experience to design, build and maintain technology…
Added 4 hours ago
Senior Electrical Engineer
Onto Innovation Inc-US
United States
Onto Innovation is seeking a Senior Electrical Engineer to design, integrate, troubleshoot, and support electrical systems for advanced semiconductor inspection and metrology products. This…
Added 4 hours ago
Computer Vision Software Engineer
BOOZ ALLEN HAMILTON INC
Dayton, OH
Job Number: R0246213 Computer Vision Software Engineer The Opportunity: As a software developer, you know that good software is more than a nice-looking interface and data. You need the…

OKX

Staff Engineer, Machine Learning & Data Infrastructure

San Jose, CA, United States

The salary range for this position is $214,666 to 321,999

Apply About employer

Applications open in a new tab.

Job details

Pay: The salary range for this position is $214,666 to 321,999
Posting language: English
Posted: Nov 4, 2026

Original job description

Who We Are

At OKX, we believe that the future will be reshaped by crypto, and ultimately contribute to every individual's freedom. OKX is a leading crypto exchange, and the developer of OKX Wallet, giving millions access to crypto trading and decentralized crypto applications (dApps). OKX is also a trusted brand by hundreds of large institutions seeking access to crypto markets. We are safe and reliable, backed by our Proof of Reserves. Across our multiple offices globally, we are united by our core principles: We Before Me, Do the Right Thing, and Get Things Done. These shared values drive our culture, shape our processes, and foster a friendly, rewarding, and diverse environment for every OK-er.

About the Opportunity

We are looking for an experienced engineer to lead the architecture and hands-on development of the ML and data infrastructure behind our risk and fraud prevention systems. You will help teams move models from experimentation into reliable production services, improve model execution performance, and build scalable data foundations for training, evaluation, and real-time decisions.

What You’ll Be Doing

• Improve model inference latency, throughput, resource utilization, and cost. Profile the full prediction path - from feature retrieval and preprocessing to model execution - and deliver measurable improvements without unacceptable model-quality regressions.

• Build reusable infrastructure for model packaging, versioning, deployment, serving, monitoring, and rollback across cloud platforms and regions.

• Design and optimize offline data pipelines on Databricks, Apache Spark, and warehouse/lakehouse technologies, addressing data modeling, partitioning, joins, shuffles, skew, incremental processing, and cost.

• Connect offline and online feature pipelines, ensuring point-in-time correctness, consistent feature definitions, freshness, lineage, and reproducible backfills.

• Evolve reliable feature services and data integrations supporting risk models and decision systems.

• Set technical direction, mentor engineers, and lead delivery across ML, data engineering, cloud infrastructure, security, and product teams.

What We Look For In You

• A record of building and operating large-scale distributed systems, ML infrastructure, or data platforms, with technical ownership across teams.

• Hands-on experience optimizing production model performance, supported by measurable improvements in latency, throughput, utilization, or cost.

• Deep understanding of Databricks and Spark, or comparable offline data platforms, including execution plans, distributed query optimization, storage layouts, data correctness, and workload cost management.

• Experience with the model lifecycle, including training/serving integration, validation, deployment, observability, and rollback.

• Strong Python, SQL, and at least one backend or systems language such as Java, Scala, Go, or C++.

• Ability to turn ambiguous requirements into clear architecture and deliver maintainable systems through hands-on engineering and technical leadership.

Nice to Haves

• SageMaker, Alibaba Cloud PAI/EAS, or comparable ML platforms.

• Inference optimization using tools such as ONNX Runtime, TensorRT, or Triton.

• Delta Lake, MLflow, feature stores, Kafka, Flink, and low-latency data stores.

• Fraud detection or financial risk systems.

• Fluency in Mandarin

Perks & Benefits

• Competitive total compensation package

• L&D programs for employees' growth and development

• Various team building programs and company events

• Wellness and meal allowances

• Comprehensive healthcare schemes for employees and dependants

• More that we love to tell you along the process!

OKX Statement:

OKX is committed to equal employment opportunities regardless of race, color, genetic information, creed, religion, sex, sexual orientation, gender identity, lawful alien status, national origin, age, marital status, and non-job related physical or mental disability, or protected veteran status. Pursuant to the San Francisco Fair Chance Ordinance, we will consider for employment qualified applicants with arrest and conviction records.

The salary range for this position is $214,666 to 321,999

The salary offered depends on a variety of factors, including job-related knowledge, skills, experience, and market location. In addition to the salary, a performance bonus and long-term incentives may be provided as part of the compensation package, as well as a full range of medical, financial, and/or other benefits, dependent on the position offered. Applicants should apply via OKX internal or external careers site.

Notice:

All official OKX vacancies are published on this website. While roles may appear on selected third-party platforms from time to time, information on other sites may be inaccurate or outdated. If in doubt, please apply directly through our official careers website.

Information collected and processed as part of the recruitment process of any job application you choose to submit is subject to OKX's Candidate Privacy Notice.

Apply About employer

Report this job