Un-0

// Overview

Un-0 is an image-generation model built on Kuramoto dynamics: it produces an image by integrating the phase dynamics of a population of coupled oscillators. There is no diffusion schedule, no adversary and no iterative denoising step — the learned parameters are the couplings between oscillators, and generation is the simulation of that system settling into a state.

It comes from Unconventional AI, a group researching dynamical systems as a computing substrate on the argument that such systems map onto analog and physical hardware, pointing toward far lower energy use per generated image than today's digital accelerators. The repository is the plain-PyTorch reference implementation of that idea, deliberately readable rather than optimised.

Two independent pipelines are trained separately and shipped together: CIFAR-10 at 32x32 across 10 classes, and ImageNet-64 at 64x64 across 1000 classes. Six pretrained checkpoints are released, named by task and oscillator count, along with the full training recipe. The published scores are FID 8.86 for cifar10/n4096 under clean-FID and FID 6.74 for imagenet64/n16384 under the OpenAI ADM evaluator — measured with different methodologies, so the two columns are not directly comparable. Training is verified on NVIDIA A100, H200 and B200 GPUs.

// What it does

• Generation by integrating coupled Kuramoto oscillator dynamics — no diffusion schedule, adversary or denoising loop
• Six released checkpoints named {task}/n{oscillators}, from a 1.3M-parameter CIFAR-10 model up to a 322M-parameter ImageNet-64 model
• from_pretrained rebuilds the architecture from the checkpoint's own config, so the right model is constructed automatically
• Full training recipes for both CIFAR-10 and ImageNet-64, single-GPU or 8-GPU DDP, with every hyperparameter exposed as a CLI flag
• Optional Weights & Biases logging and clean-FID evaluation, pulled in only as separate dependency groups
• Device selection defaults to auto, preferring CUDA, then Apple Silicon MPS, then CPU

// Getting started

Requires Python 3.11 or newer and PyTorch 2.11 or newer; a CUDA GPU is recommended. The project uses uv for dependency management.

uv sync --group dev

import torch
from un0 import ConditionalImplicitKuramotoGenerator
from torchvision.utils import save_image

model = ConditionalImplicitKuramotoGenerator.from_pretrained("cifar10/n4096")

class_ids = torch.tensor([0, 1, 2, 3, 4])      # one image per listed class
images = model.sample_images(class_ids)         # (5, 3, 32, 32) in [0, 1]
save_image(images, "samples.png", nrow=5)

uv run python un0/inference.py \
    --pretrained imagenet64/n16384 \
    --classes 0 1 207 \
    --samples-per-class 8 \
    --output samples/imagenet64.png

uv run python un0/train_cifar10.py --checkpoint-dir checkpoints/cifar10

uv run torchrun --nproc_per_node=8 un0/train_cifar10.py \
    --checkpoint-dir checkpoints/cifar10 \
    --wandb-project <your-wandb-project>

Commands and code are distilled from the project's own documentation — always check the official repo for the latest.

// When to use it

• Experiment with a generative model whose compute is dynamical-system simulation rather than a neural denoising loop
• Study how sample quality scales with oscillator count using the six released checkpoints as a ladder
• Reproduce the published CIFAR-10 and ImageNet-64 results with the included training recipes and evaluation methodology
• Prototype workloads aimed at analog or physical computing substrates, where oscillator dynamics map onto the hardware

// How Un-0 compares

Un-0 alongside other open-source image generation tools AI/TLDR tracks, ranked by GitHub stars.