A collaborative virtual lab built by AI/ML enthusiasts from around the internet.
We publish research at top venues, build infrastructure we rely on daily, and share everything openly. 100% Open Access.
QQ: 881970728 | Discord1: https://discord.gg/tPBsKDyRR5 | Discord2: https://discord.gg/jHHbTaHV3a
Kohaku Lab is a virtual-first research lab — no office, no corporate structure, just people who love building AI/ML tools and doing research together across time zones.
We publish at top venues (LyCORIS at ICLR 2024, TIPO at ICLR 2026), build the infrastructure we need (an agent framework, model hosting, cluster management, experiment tracking), create databases and ML utilities, and even play with AI-powered board games. The philosophy is simple: scratch your own itch.
Led by KohakuBlueLeaf (Shih-Ying Yeh) and joined by contributors worldwide, we work across the full stack of AI — from neural network research and GPU/FPGA kernels to production infrastructure, agent systems, and databases.
Projects under the Kohaku-Lab org are the lab's collaborative work; the rest live on KohakuBlueleaf, the founder's account.
| Project | Description | Stack |
|---|---|---|
| LyCORIS | Lora beYond Conventional methods — comprehensive PEFT library for neural networks (ICLR 2024, 2.5k★) | Python PyTorch |
| TIPO / KGen | Text-to-Image with Text Presampling for Prompt Optimization (ICLR 2026) | Python LLM |
| HDM | Home-made Diffusion Model — training quality T2I models on consumer hardware | Python Diffusion |
| HakuTPU | AI accelerator implementation on Xilinx FPGAs | Verilog |
| Project | Description | Stack |
|---|---|---|
| KohakuRAG | Simple hierarchical RAG framework — tree-structured indexing over single-file SQLite storage | Python SQLite |
| full-self-coding_HCI | Experiment: no prompts, no instructions, no plans — 100–1000 AI agents coding in parallel | TypeScript |
| Project | Description | Stack |
|---|---|---|
| KohakuHub | Self-hosted HuggingFace alternative — model hosting, versioning, S3 storage, org management | Python FastAPI Vue 3 |
| KohakuRiver | Lightweight cluster manager — turns a small fleet into one powerful computer using Docker | Python |
| KohakuBoard | High-efficiency local/self-hosted ML experiment tracking & visualization | Python Vue 3 |
| Project | Description | Stack |
|---|---|---|
| KohakuVault | Single-file key-value store on SQLite — storage backend for KohakuRAG and KohakuTerrarium sessions | Rust Python |
| Project | Description | Stack |
|---|---|---|
| PixelOE | Detail-oriented pixelization via contrast-aware outline expansion | Python CV |
| HakuBooru | Text-image dataset maker for anime-style images | Python |
| IGS | Image Gaussian Splatting | Python |
| OptimFactory | Optimizer utils with frontier hyperparameter transfer rules | Python PyTorch |
| SenseCraft | Unified perceptual feature loss framework | Python PyTorch |
| AnySchedule | Hyperparameter schedule composer for PyTorch | Python |
| Project | Description | Stack |
|---|---|---|
| KohakuBoardGame | Board game engine with NNUE neural network evaluation and PVS search | C++ Python |
Passion — Build what excites you, work on what interests you
Joy — If it's not fun, take a break. Happiness first
Evolution — New tech for new projects. Let old projects be
Openness — Good tools should be accessible to everyone
We're always open to new collaborators. Whether you're a researcher, developer, student, or just passionate about AI — there's a place for you.
- Website: kohaku-lab.org
- Discord: Join our server
- Contact: kohaku-lab.org/contact
- GitHub: github.com/Kohaku-Lab
