Kohaku BlueLeaf
Shih-Ying Yeh 葉適穎
CS master's student at National Tsing Hua University in Taiwan, also ML/NN engineer and core developer at Comfy Org. Published at ICLR 2024 (LyCORIS), ICLR 2026 (TIPO, ImagenWorld) and ECCV 2026 (ReactVAU).
I research and build AI systems end-to-end: from neural network training and GPU kernels to infrastructure, databases and agent systems. I also design compiler/IR/programming language for different projects, and design my own AI accelerator chip on my own accelerator framework (KohakuAccel/KohakuTPU). Open-source projects got over 7k stars on GitHub and 6M downloads on PyPI, models got over 3M downloads on HuggingFace in total.
Leading Kohaku Lab, a virtual open-source research lab. Currently working on agent framework (KohakuTerrarium), FPGA AI accelerator (KohakuTPU), black-box optimization (KohakuHPO) and VAE/AE/VQ-VAE architectures for generative modeling. Everything we build is 100% open access.
"AI art should look like AI, not humans."
Recent Projects
LyCORIS
Lora beYond Conventional methods. Comprehensive PEFT library for neural networks, published at ICLR 2024.
TIPO
Text to Image with Text Presampling for Prompt Optimization. Published at ICLR 2026, related models got over 3M downloads.
HDM
Home-made Diffusion Model. Training quality text-to-image models on consumer-level hardware. Tech report on arXiv.
KohakuTerrarium
General purpose AI agent framework with built-in tools, sub-agents, persistent sessions, TUI and web UI. Comes with kt CLI and package marketplace.
KohakuTPU [WIP]
AI accelerator implementation with Xilinx FPGA. Built on KohakuAccel, an open platform for building FPGA accelerators.
PixelOE
Detail-Oriented Pixelization based on Contrast-Aware Outline Expansion.
Recent Publications
ReactVAU: A Slow-Fast Decoupled Framework for Streaming Video Anomaly Understanding
TIPO: Text to Image with Text Presampling for Prompt Optimization
ImagenWorld: Stress-Testing Image Generation Models with Explainable Human Evaluation on Open-ended Real-World Tasks
Recent Career
ML/NN Engineer (Core Dev)
@ Comfy OrgCore developer of ComfyUI, tightly collaborated with comfy team to improve the usability and performance of Comfy UI eco system. The whole weight adapter system (LoRA/LoHa/LoKr/DoRA/OFT/BOFT) is created and maintained by me. Also working on lora trainer nodes, dataset processing nodes and FP4/FP8/FP16 training support.
Open Source Contributor
@ Kohaku LabDeveloped and maintained open-source projects related to computer vision and AI. Included personal project such as LyCORIS, TIPO, PixelOE, HakuLatent and more, and community projects such as stable-diffusion-webui (core dev), sd-scripts, ComfyUI (core dev) and more. Founded Kohaku Lab in Apr 2025, a virtual open source research lab for KohakuTerrarium, KohakuRAG, KohakuHub, KohakuRiver and more.
ML Engineer (Part-time)
@ AppleCollaborated with research team in Apple Inc. to explore the possibility of new form dataset. The work "Graph-Based Captioning: Enhancing Visual Descriptions by Interconnecting Region Captions" has been published on Arxiv and Apple official github.
