AI Architecture Team Lead
Primary stack
Nice to have's
Job description
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
© HIRE.UA. This job description was sourced from the employer's public career page. TheJob is not the employer — we index the posting and route candidates to the source. All content rights and hiring decisions belong to the employer.
HIRE.UAis a large-scale recruiting organization in Ukraine. Our goal is to bring together qualified professionals to open up new horizons and opportunities. Why settle for good when you can get the best?HIRE.UAis your personal outsourcing recruiting solution expert. With us, it is easy to create and implement HR programs specifically for your business.HIRE.UAhandles any staffing issues. We do not offer a candidate for a position, we offer an experienced specialist to achieve your goals. Effective attraction of the best candidates to the organization is guaranteed by us.
