Official PyTorch Implementation for the "Recovering the Pre-Fine-Tuning Weights of Generative Models" paper (ICML 2024).

Official PyTorch Implementation for the "Unsupervised Model Tree Heritage Recovery" paper (ICLR 2025).

A collection of weight space learning including papers, codes, and datasets.

Awesome papers on weight-space learning

Code Repository for the ICML 2024 paper: "Towards Scalable and Versatile Weight Space Learning".

Code Repository for the NeurIPS 2022 paper: "Hyper-Representations as Generative Models: Sampling Unseen Neural Network Weights".

Official PyTorch Implementation for the "Learning on Model Weights using Tree Experts" paper (CVPR 2025).

An official implementation of ProbeGen

Code Repository for the CVPR 2026 paper: "Learning Geospatial Representations from Models, not Data".

AAAI'26 Oral: "WeightFlow: Learning Stochastic Dynamics via Evolving Weight of Neural Network"

Weight-Space Linear Recurrent Neural Networks

Website for the ICLR 2025 Weight Space Learning workshop.

SANE backbone trained on multiple model zoos.

[ICLR 2026] Weight Space Representation Learning on Diverse NeRF Architectures