8 projects
torchebm
Simulation-free, GPU-first generative modeling in PyTorch: composable primitives for scalable, stable training of modern energy-based models, diffusion, flow matching, and Schrödinger bridges.
worldkernels
GPU-first world model simulation engine for serving learned world models (DiT, VAE) as interactive sessions
transformerx
TransformerX is a python library for building transformer-based models using ready-to-use layers.
sciart
emgraph
A Python library for knowledge graph representation learning (graph embedding)
cvit
tase
bigraph
Python package for link prediction in bipartite graphs and networks
