Engineer, Supercomputing & Distributed Systems

Supercomputing / AI Infra at Krea

We build and operate the infrastructure for Krea's research and inference. Distributed training, 1000+ K8s GPU clusters, petabyte scale data pipelines, etc. We build a lot of this from scratch - custom distributed datastores, job orchestration systems, and streaming pipelines that replace tools like Kafka and Ray for modern AI workloads at scale.

Example Projects: Distributed Data Systems

• Design multi-stage pipelines that turn petabytes of raw data into clean, annotated datasets
• Run classification models on billions of images
• Deploy and combine LLMs to caption massive multimedia data

Gpu Infrastructure

• Manage distributed training and inference on 1000+ GPU Kubernetes clusters
• Solve orchestration and scaling for large-scale GPU job processing
• Scale workloads and research between clusters in multiple datacenters

Distributed Training

• Profile and optimize dataloaders streaming thousands of images per second
• Profile and debug InfiniBand networking on huge training runs
• Build fault tolerance systems for large-scale pretraining
• Collaborate with researchers on evolving RL infrastructure

Applied Ml Pipelines

• Find clean scenes in millions of videos using distributed shot-boundary detection
• Customize and train models to filter billions of images for questions like "is this a screenshot?"
• Build the systems that bridge raw cluster capacity and research output

Who We're Looking for

Systems people. If you've read a blog post about InfiniBand debugging or building a custom distributed database and thought "I want to do that" - this is that team.

You'll spend your time working heavily with Python, Kubernetes, Torch, and data tools like DuckDB, Arrow, etc. It's OK if you don't have K8s or ML experience - the main thing we hire for is an intuition for distributed systems, and a great mental model of how systems interact and function under different conditions.

Strong Candidates May Have Experience with

• Python, PyArrow, DuckDB, SQL, massive relational databases, PyTorch, Pandas, NumPy…
• Kubernetes
• Designing and implementing large-scale ETL systems
• Fundamental knowledge of containerization, operating systems, file-systems, and networking
• Distributed systems design
• Distributed training systems (NCCL, InfiniBand, RDMA)
• Streaming and event processing systems (Kafka, Pulsar, or similar)
• PyTorch internals, custom dataloaders, and training infrastructure

What We Offer

• Team: Work alongside a world-class team building the future of AI creative tooling
• Impact: Significant scope and company-wide impact
• Competitive compensation: generous salary & equity packages
• Health & wellness: 100% health & 99% dental/vision insurance premiums covered for employees, health FSA accounts, & long-term disability coverage
• Time off: Flexible PTO policy
• Financial planning: 401k with a 4% company-sponsored match
• Meals in the office: breakfast, lunch, dinner - you name it, we'll cover it
• Transit: Ubers covered to & from the office
• Sponsorship: We're open to sponsoring international visas where we can (e.g., STEM OPT, OPT, H-1B, O-1, E-3).
• And more!

Please note the above benefits & perks are for full-time employees