Krea
United States
New
Growth Engineer
- 3D
- AI
- Android
- AWS
K
AI Summary
Train and optimize large-scale diffusion models for image and video generation on distributed GPU clusters. Implement and debug complex distributed training strategies including FSDP, TP, and SP to ensure efficiency and reliability. Requires strong proficiency in PyTorch, distributed systems, and a proven track record in training generative models at scale.
Key Highlights
Train diffusion models for image and video generation on large GPU clusters
Optimize and profile large distributed training runs across various parallelism strategies
Debug distributed training errors and implement fault tolerance solutions
Searching for Development & Programming roles that provide visa sponsorship? Connect with international employers through Development & Programming Jobs with Visa Sponsorship opportunities actively seeking talented professionals.
Key Responsibilities
Train diffusion models for image and video generation on large GPU clusters
Fully optimize and profile large distributed training runs across model architectures, kernels, data loading, memory constraints, and communication
Implement and improve various distributed training strategies including FSDP, CP, SP, TP, and EP
Continuously improve model quality and reliability through data, model architecture, training pipeline, structuring experiments, and eval design
Debug distributed training errors and implement fault tolerance solutions, identifying bad GPU, NVLink, Infiniband (IB) components as well as monitoring numerical errors and NCCL issues
Ablate different architecture, attention, optimizer, data, and algorithmic choices to reliably improve efficiency and performance of our models
Technical Skills Required
Explore our comprehensive directory of visa sponsorship jobs from employers worldwide who are ready to sponsor talented international professionals.
Benefits & Perks
100% health & 99% dental/vision insurance premiums covered
401k with a 4% company-sponsored match
Meals in the office covered
Ubers covered to & from the office
Flexible PTO policy
Interested in opportunities specifically in United State? Discover our dedicated Visa Sponsorship Jobs in United State page featuring roles from top employers in this location.
Nice to Have
Publications or open-source contributions in image or video models at scale
Good knowledge of low precision training / inference in FP8, NVFP4, and MXFP8
Keeping up with developments in LLM, VLM, representation learning, and robotics research
Job Description
At Krea, we are building next-generation AI creative tools.We're dedicated to making AI intuitive and controllable for creatives - our mission is to build tools that empower human creativity, not replace it. We believe AI is a new medium that allows us to express ourselves through various formats - text, images, video, sound, and even 3D. We're building better, smarter, and more controllable tools to harness this medium. We recently took this a step forward with the launch of Krea 2, our first foundation model, built completely from scratch for aesthetic diversity and stylistic control.
Want the full job description?
Read the complete details on LinkedIn, the original posting.
Continue on LinkedIn
This is a short excerpt. All rights to the full description belong to its original publisher.
Keep exploring
Similar Jobs
Explore other opportunities that match your interests
Rnb Consultancy
United States
New
Agent Systems Engineer – AI Model Routing
- AI
- Golang
- Java
- Node.js
Hidden company
Log in to see location
Premium
Senior Backend Engineer, AI Gateway
Log in free to see the details
Visas, Work Permits & Legal Insights
Stay informed about visa types, work permits, and immigration updates for international professionals.
Visa Sponsorship Jobs for Pakistanis Abroad (2026 Guide)
A visa-sponsored job for Pakistanis is when a company in another country agrees to support your work visa. If you don’t have that...
Read Article
Dubai Visa Sponsorship Jobs for Indians (2026 Guide)
Landing a Dubai-based position with full visa sponsorship for Indian candidates is possible if you know how to play the game. Thou...
Read Article
Companies in the Netherlands That Sponsor Visas (2026 List)
Securing a visa sponsorship job in the Netherlands is a popular choice for many international job seekers around the world. This p...
Read Article
Companies That Sponsor Visas in Canada (2026 List)
Securing a visa sponsorship job in Canada is a popular choice for many international job seekers around the world. This process ca...
Read ArticleSubscribe to our newsletter
New visa sponsorship, relocation and remote tech jobs, delivered to your inbox.