gugarosa/learnergy

🍓 Simulation-free, GPU-first generative modeling in PyTorch ⚡ Composable primitives for scalable, stable training of modern EBMs, diffusion, flow matching, and Schrödinger bridges.

Machine Learning for High Energy Physics.

Generative Flow Networks - GFlowNet

Tensors and Dynamic neural networks in Python with strong GPU acceleration

A deep learning library built from scratch with complex neural networks examples built on top for learning purposes.

A Python toolkit for Reservoir Computing and Echo State Network experimentation based on pyTorch. EchoTorch is the only Python module available to easily create Deep Reservoir Computing models.

Scenic: A Jax Library for Computer Vision Research and Beyond

TorchMultimodal is a PyTorch library for training state-of-the-art multimodal multi-task models at scale.

solo-learn: a library of self-supervised methods for visual representation learning powered by Pytorch Lightning