Senior Data Engineer AI Infrastructure

Unchain Data

United States

On-site

USD 120,000 - 160,000

Full time

14 days+

Get more replies from employers

Send a job-specific resume in minutes.

Job summary

Unchain Data is seeking an experienced engineer to own GPU and accelerator clusters, crucial for AI workloads. You will work closely with teams to build efficient and reliable AI infrastructure while optimizing performance and cost. Ideal candidates will have significant experience in infrastructure engineering, production environments, and a proficiency in Python.

Join a high-impact team dedicated to advancing AI capabilities in a fast-paced, innovative environment.

Qualifications

  • 5+ years of experience with GPU clusters or accelerator-backed infrastructure.
  • Hands-on experience with scheduling, orchestration, and cost optimization.
  • Strong fundamentals in Linux, networking, and storage.

Responsibilities

  • Own and operate GPU and accelerator clusters.
  • Design infrastructure for running models locally on GPUs.
  • Partner with ML engineers to optimize training and inference workflows.

Skills

Infrastructure engineering experience
GPU compute
Machine Learning infrastructure
High-performance computing
Python proficiency
Systems engineering fundamentals
Distributed systems
Production debugging

Tools

Kubernetes
TensorRT
Triton Inference Server
CUDA

Job description

Building the Future of Open Finance

Payward – the parent company behind Kraken, NinjaTrader, Breakout, xStocks, Payward Services and CF Benchmarks – has spent the last 15 years building one of the most modern and globally accessible financial infrastructure platforms in the industry, built to advance an open, global financial system.

Founded in 2011, Kraken is one of the world’s longest‑standing crypto platforms, trusted by over 10 million individuals and institutions across the globe. It offers spot trading, margin, futures, staking, and OTC services, with products built for both individual investors and institutional clients.

Kraken is building a dedicated AI Compute and Infrastructure team to power the next generation of model training, inference, evaluation, and experimentation across the exchange. This team sits within engineering leadership and owns the infrastructure layer that lets Kraken run AI workloads with control, speed, reliability, and cost discipline.

The team is responsible for GPU and accelerator infrastructure, cluster operations, scheduling, model serving, observability, capacity planning, and cost‑efficient compute at scale. This is the backbone that allows Kraken to train, serve, evaluate, and iterate on AI systems in‑house where it matters for privacy, latency, reliability, cost, or product differentiation.

You will join a small, senior, high‑impact team working directly with AI/ML researchers, platform engineers, security teams, and product teams. The mandate is simple: make Kraken’s AI ambitions real by building compute infrastructure that is fast, dependable, efficient, and production‑grade.

The opportunity
  • Own and operate GPU and accelerator clusters used for training, inference, evaluation, and experimentation, including drivers, runtimes, kernels, device plugins, node configuration, scheduling primitives, and workload isolation.
  • Design infrastructure that enables Kraken teams to run models locally on GPUs where it is strategically and economically preferable, reducing unnecessary dependency on external providers and containing compute costs.
  • Build and improve scheduling, orchestration, placement, quota management, and utilization systems across heterogeneous accelerator environments.
  • Optimize inference pipelines for latency, throughput, reliability, memory efficiency, and cost using frameworks such as vLLM, Triton Inference Server, TensorRT, or equivalent serving stacks.
  • Partner with ML engineers and researchers to remove bottlenecks in training, evaluation, batch inference, online inference, deployment, and production debugging workflows.
  • Build observability for GPU utilization, memory pressure, queue depth, saturation, token throughput, request latency, failed workloads, capacity pressure, and spend.
  • Drive reliability, incident response, alerting, runbooks, and post‑incident improvements for always‑on AI compute infrastructure.
  • Evaluate and integrate new hardware, cloud instance families, specialized accelerators, runtimes, schedulers, and serving frameworks as the AI infrastructure landscape evolves.
  • Build tooling that makes GPU usage visible, accountable, and easier for internal teams to consume without needing to become infrastructure experts.
  • Contribute to long‑term architecture decisions that balance performance, cost efficiency, scalability, operational simplicity, and production safety.
What You Bring
  • 5+ years of infrastructure engineering experience, with significant time spent on GPU compute, ML infrastructure, distributed systems, high‑performance computing, or large‑scale production platforms.
  • Hands‑on experience operating GPU clusters or accelerator‑backed infrastructure in production or production‑like environments, including scheduling, orchestration, utilization monitoring, and cost optimization.
  • Strong systems engineering fundamentals across Linux, networking, storage, containers, Kubernetes, distributed runtimes, and production debugging.
  • Experience with ML serving frameworks such as vLLM, Triton Inference Server, TensorRT, TorchServe, KServe, Ray Serve, or equivalent systems.
  • Proficiency in Python for infrastructure automation, tooling, debugging, integration, and operational workflows.
  • Practical understanding of performance tradeoffs across batching, concurrency, memory usage, GPU utilization, model size, latency, throughput, availability, and cost.
  • Track record of optimizing compute costs while maintaining clear performance, reliability, and availability expectations.
  • Experience building observable systems with useful metrics, logs, traces, dashboards, alerts, and incident workflows.
  • Comfortable working in high‑stakes, always‑on environments where uptime, throughput, correctness, and operational discipline are critical.
  • Clear communicator who can translate infrastructure tradeoffs for researchers, product teams, platform engineers, security stakeholders, and engineering leadership.
Nice to haves
  • Experience at a frontier AI lab, hyperscaler, high‑frequency trading firm, research platform, or high‑scale ML organization.
  • Familiarity with custom silicon or specialized accelerators such as TPUs, AWS Trainium, Gaudi, or similar platforms.
  • Background in capacity planning, procurement input, reserved capacity strategy, cloud accelerator economics, or GPU fleet cost management.
  • Experience with distributed training frameworks such as DeepSpeed, Megatron‑LM, FSDP, Ray, or equivalent systems.
  • Experience debugging CUDA, NCCL, kernel, driver, runtime, memory, networking, or low‑level performance issues.
  • Experience with Rust, C++, Go, CUDA, or other systems languages used for performance‑critical infrastructure.
  • Crypto, financial services, trading infrastructure, or security‑sensitive production infrastructure experience.

Unless a specific application deadline is stated in the job posting, applications are accepted on an ongoing basis.

Please note, applicants are permitted to redact or remove information on their resume that identifies age, date of birth, or dates of attendance at or graduation from an educational institution.

We consider qualified applicants with criminal histories for employment on our team, assessing candidates in a manner consistent with the requirements of the San Francisco Fair Chance Ordinance.

Payward is powered by people from around the world and we celebrate the diverse talents, backgrounds, contributions, and unique perspectives that everyone brings to the table. We hire based on merit, seeking out people with the right abilities, knowledge, and skills for the job. We encourage you to apply for roles where you don’t fully meet the listed requirements, especially if you’re passionate or knowledgeable about crypto.

We may ask candidates to complete job‑related skills or work‑style assessments as part of our hiring process. These assessments evaluate competencies relevant to the role and are applied consistently across candidates for similar positions. Results are considered alongside experience and interviews, and are not the sole basis for any employment decision.

As an equal opportunity employer, we don’t tolerate discrimination or harassment of any kind, whether based on race, ethnicity, age, gender identity, citizenship, religion, sexual orientation, disability, pregnancy, veteran status, or any other protected characteristic as outlined by federal, state, or local laws.

Candidate Privacy Notice

Get your free, confidential resume review.
or drag and drop your file here.
Similar jobs

Similar jobs worth comparing

Infrastructure Engineer - Core Infrastructure
Infrastructure Engineer - Core Infrastructure

Kraken • Peru (IL)

On-site
USD 90,000 - 130,000
Senior Software Engineer - AI Enablement HR
Senior Software Engineer - AI Enablement HR

Kraken International • Northern (KY)

Hybrid
USD 150,000 - 190,000
Senior Software Engineer - AI Enablement Compliance
Senior Software Engineer - AI Enablement Compliance

Kraken International • Northern (KY)

Hybrid
USD 170,000 - 250,000
Infrastructure Engineer - Kubernetes Platform
Infrastructure Engineer - Kubernetes Platform

Kraken • Peru (IL)

On-site
USD 120,000 - 180,000
Senior Data Engineer - Agents Systems
Senior Data Engineer - Agents Systems

Unchain Data • United States

On-site
USD 100,000 - 130,000
Senior AI Engineer - Growth
Senior AI Engineer - Growth

Kraken International • United States

On-site
USD 120,000 - 180,000
Senior Internal Auditor, Technology
Senior Internal Auditor, Technology

Kraken International • Northern (KY)

Hybrid
USD 100,000 - 160,000
Senior Internal Auditor, Technology
Senior Internal Auditor, Technology

Kraken • United States

On-site
USD 73,000 - 145,000
Senior Internal Auditor, Technology
Senior Internal Auditor, Technology

Kraken • San Francisco (CA)

On-site
USD 180,000 - 240,000
Sr. Data Analyst, Data & Audit Readiness
Sr. Data Analyst, Data & Audit Readiness

Kraken • United States

On-site
USD 110,000 - 211,000