Krea AI

Machine Learning Engineer at Krea AI, focusing on large-scale training of foundation diffusion models for image and video generation, including controllability modules. The role involves developing novel research techniques and implementing them in production, optimizing data pipelines, and conducting experiments on HPC clusters.

ML Engineer to architect and build Krea's personalization and recommendation systems from scratch, focusing on user taste, content curation, and adapting generative models to individual aesthetics. This role involves designing algorithms, building curated feeds, and contributing to personalized image generation research, taking systems from research to production.

The role focuses on building and operating the infrastructure for AI research and inference, including distributed training, large GPU clusters, petabyte-scale data pipelines, custom distributed datastores, job orchestration systems, and streaming pipelines. It involves designing multi-stage data pipelines, managing distributed training and inference on large GPU clusters, scaling workloads, and optimizing dataloaders and networking for large training runs. The role requires strong systems thinking and experience with distributed systems, Python, Kubernetes, and data tools.