8 projects
torchebm
Components and algorithms for energy-based models
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