AI4Science-WestlakeU/t_scend
This repo is the code for T-SCEND, a novel framework that significantly improves diffusion model’s reasoning capabilities with better energy-based training and scaling up test-time computation.
This project helps AI researchers and machine learning engineers significantly improve the reasoning capabilities of diffusion models. It takes existing diffusion models and through advanced training and scalable computation, outputs models that can solve complex tasks, like larger mazes or Sudoku puzzles, with higher accuracy even when trained on simpler versions. This is designed for those pushing the boundaries of AI problem-solving.
Use this if you are developing or experimenting with diffusion models and need to enhance their ability to reason and generalize to more complex problems than their training data.
Not ideal if you are looking for a ready-to-use application to solve specific problems without delving into model training and research.
Stars
26
Forks
1
Language
Python
License
MIT
Category
Last pushed
Oct 19, 2025
Commits (30d)
0
Get this data via API
curl "https://pt-edge.onrender.com/api/v1/quality/diffusion/AI4Science-WestlakeU/t_scend"
Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.
Higher-rated alternatives
quantgirluk/aleatory
📦 Python library for Stochastic Processes Simulation and Visualisation
blei-lab/treeffuser
Treeffuser is an easy-to-use package for probabilistic prediction and probabilistic regression...
TuftsBCB/RegDiffusion
Diffusion model for gene regulatory network inference.
yuanchenyang/smalldiffusion
Simple and readable code for training and sampling from diffusion models
chairc/Integrated-Design-Diffusion-Model
IDDM (Industrial, landscape, animate, latent diffusion), support LDM, DDPM, DDIM, PLMS, webui...