Engineer, Supercomputing & Distributed Systems

Krea is an active generative-AI creative platform for generating, editing, and enhancing images, video, and 3D content.

San Francisco, United States
About Krea

Krea provides browser-based AI creative tools and a developer API. Its offerings include image and video generation, enhancement and editing tools, 3D generation, and Krea 2, its in-house image foundation model.

View jobs by Krea

Skills

Candidate Availability

Required and preferred rules are kept separate and reflect the wording in the original posting.

About the Role

You will build and operate distributed infrastructure for research and inference. You will develop large-scale data pipelines, GPU orchestration, distributed training systems, and applied machine-learning workflows.

Requirements

  • Distributed systems design
  • Python
  • Fundamental knowledge of containerization, operating systems, file systems, and networking

Responsibilities

  • Design multi-stage pipelines for petabyte-scale datasets
  • Run classification models on billions of images
  • Deploy and combine LLMs to caption multimedia data
  • Manage distributed training and inference on GPU Kubernetes clusters
  • Solve orchestration and scaling for GPU job processing
  • Optimize data loaders and debug InfiniBand networking
  • Build fault-tolerance systems for pretraining
  • Collaborate on reinforcement-learning infrastructure
  • Build distributed systems for video and image-processing workflows

Benefits

  • Health insurance with 100% employee premium coverage
  • Dental and vision insurance with 99% employee premium coverage
  • Health FSA accounts
  • Long-term disability coverage
  • Flexible PTO
  • 401k with a 4% company-sponsored match
  • Office meals
  • Uber transit to and from the office
  • International visa sponsorship where possible