← All roles

Engineer, Supercomputing & Distributed Systems

Krea· 01. Engineering
Apply on company site
San Francisco$0Full-timePosted 6mo ago

About the role

Engineering role focused on building and operating Krea's AI research and inference infrastructure, including distributed training, large-scale GPU clusters, petabyte-scale data pipelines, and custom data/streaming systems.

Responsibilities

  • 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
  • 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
  • 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
  • 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

Qualifications

  • Strong intuition for distributed systems and how systems interact under different conditions
  • Experience with Python, Kubernetes, Torch, and data tools like DuckDB, Arrow, etc.
  • Experience with K8s or ML is not required if distributed systems intuition is strong

Skills

Required:PythonKubernetesTorch / PyTorchDistributed systems design
Nice to have:Data tooling (DuckDB, Arrow, SQL)

Benefits

  • 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
  • Flexible PTO policy
  • 401k with a 4% company-sponsored match
  • Meals in the office: breakfast, lunch, dinner
  • Transit: Ubers covered to & from the office
  • Sponsorship: open to sponsoring international visas where possible (e.g., STEM OPT, OPT, H-1B, O-1, E-3)
Last updated Aug 5, 2026. Interested? Apply on the company site.

About Krea

The most powerful AI suite for Creatives.

Generate, edit, and enhance images, videos, and 3D assets with Krea's creative AI suite. Start for free with real-time tools, powerful models, and collaborative workflows.

AI Image GenerationAI Video GenerationAI 3D GenerationImage EnhancementsVideo EnhancementsFinetuningAsset Management
Company profileWebsite

More open roles at Krea