AI Architecture Team Lead

About the Position

We are seeking an AI Architecture Team Lead to join our innovative AI company, dedicated to building the next generation of hyper-realistic image and video generation systems. In this role, you will lead the architecture and development of AI-driven systems, mentor a team of developers, and ensure our platform scales to support millions of users globally.

Responsibilities

• Lead the architecture and development of AI-driven image and video generation systems, optimized for scale and stability.
• Write and review clean, efficient, production-grade code (Python, CUDA, PyTorch, etc.).
• Manage a small, high-performing development team, setting clear priorities, delegating tasks, and overseeing progress.
• Maintain comprehensive documentation, including PRDs, technical specs, and API references.
• Define and execute the technical roadmap for scaling to a million+ user environment.
• Conduct daily syncs, mentor developers, and ensure code quality and team alignment.
• Collaborate closely with product leadership to plan future system evolution and user-facing innovation.

Requirements

• Comfortable working with systems that include NSFW content generation features.
• 5+ years of experience in AI/ML development, with a strong background in image or video generation (Diffusion, GANs, or transformer-based models).
• Strong command of Python, CUDA, and GPU optimization.
• Experience deploying AI models in production environments with high concurrency and scaling demands.
• Familiarity with frameworks such as Stable Diffusion, ComfyUI, PyTorch, or TensorFlow.
• Understanding of microservice architecture, API development, and containerized deployment (Docker/Kubernetes).
• Proven experience leading development teams, managing workflows, sprints, and documentation.
• Ability to think strategically while staying deeply involved in technical execution.

Nice to Have

• Experience training or fine-tuning LoRAs, or working on text-to-video or video2video pipelines.
• Previous contributions to ComfyUI custom nodes or plugin ecosystems.
• Familiarity with distributed inference systems and cloud GPU orchestration.
• Experience building AI SaaS products or developer platforms with high user volumes.

We Offer

• Competitive pay up to $11,600 USD.
• Direct influence on architecture decisions for a platform that’s scaling globally.
• Work with a proven leadership team experienced in building multi-8-figure AI companies.
• Full autonomy, clear communication, and a product-first culture.

About the Company

We’re an established AI company building the next generation of hyper-realistic image and video generation systems, used to create digital models and virtual personas at scale. Our platform already has active users and a growing base of creators leveraging these tools daily. We’re currently building P2 of our product, along with a completely new UI/UX designed to make AI model creation accessible, powerful, and scalable for millions of users.

Contact Information

• Headquarter address: BC Parus, Kyiv
• Support E-mail: [email protected]
• Contact phone: +38 063 135 4725