Skip to main content

Our Projects

Explore our open-source contributions to the AI/ML community. All projects are freely available and built with passion.

LyCORIS

LyCORIS

Comprehensive PEFT Library (ICLR 2024)

Lora beYond Conventional methods. Advanced parameter-efficient fine-tuning techniques for neural networks, published at ICLR 2024.

ResearchPythonNeural Network
View Details
HDM

HDM

Home-made Diffusion Model

Home-trained text-to-image diffusion model. Demonstrating that high-quality generative models can be trained on accessible hardware.

ResearchPythonNeural Network
View Details

TIPO

Text-to-Image Prompt Optimization (ICLR 2026)

A system for prompt generation and optimization to improve text-to-image model performance. Published at ICLR 2026.

ResearchPythonNeural Network
View Details

KohakuHub

Self-Hosted Model Hosting Platform

A self-hosted HuggingFace alternative with model hosting, versioning, and collaboration features. Built with FastAPI and Vue 3.

InfrastructurePythonFullstack
View Details

KohakuRiver

Lightweight Cluster Manager

Turns your small fleet of nodes into one powerful computer, using Docker for environment consistency without enterprise orchestration overhead.

InfrastructurePython
View Details
KohakuBoard

KohakuBoard

ML Experiment Tracking

High-efficiency local/self-hosted ML experiment tracking and visualization system. A lightweight alternative to W&B.

InfrastructurePythonML Ops
View Details

KohakuVault

Single-File Key-Value Store

Simple single-file key-value store database built on SQLite. Lightweight, portable, and designed for embedding in other applications.

DatabaseRust
View Details

KohakuRAG

Hierarchical RAG Framework

Domain-agnostic retrieval-augmented generation framework with hierarchical document indexing, context-aware search, and single-file SQLite storage.

AI/LLMPython
View Details

KohakuTerrarium

Universal Agent Framework

Async-first Python agent framework for building autonomous systems. YAML-driven configuration with nested sub-agents and channel-based coordination.

AI/LLMPython
View Details

IGS

Image Gaussian Splatting

GPU-accelerated Image Gaussian Splatting implementation for efficient image representation and rendering.

KernelsPython
View Details

KoVec

Vector Operations Library

Optimized vector operations library for high-performance numerical computing.

KernelsPython
View Details

PixelOE

Detail-Oriented Pixelization

Contrast-aware outline expansion for detail-oriented pixelization. Converts images to pixel art while preserving important details.

ML UtilsPythonComputer Vision
View Details

OptimFactory

Optimizer Collection for PyTorch

Collection of optimizer implementations for PyTorch with a unified interface for easy experimentation.

ML UtilsPython
View Details

SenseCraft

Unified Perceptual Loss

Unified perceptual loss implementation for training neural networks with perceptual quality objectives.

ML UtilsPython
View Details

AnySchedule

Hyperparameter Scheduling

A composer tool for hyperparameter scheduling in PyTorch. Flexible schedule composition for learning rates and other parameters.

ML UtilsPython
View Details

S3impleClient

S3-Compatible Storage Client

Lightweight S3-compatible storage client for simple object storage operations.

ML UtilsPython
View Details

KohakuBoardGame

AI Board Game Engine

Multi-game board game engine with NNUE neural network evaluation and PVS search. Supports MiniChess, MiniShogi, and Gomoku.

FunC++PythonNeural Network
View Details