Infinity ∞ : Scaling Bitwise AutoRegressive Modeling for High-Resolution Image Synthesis

Batch computation of the linear assignment problem on GPU.

Empowering deep neural quantum states through efficient optimization

Package for performing DQMC simulations of Hubbard and Electron-Phonon Models

A concise but complete full-attention transformer with a set of promising experimental features from various papers

Random symmetric initialization of crystals

Generic template to bootstrap your PyTorch project.

Riemannian geometry in JAX

web-based physical simulation written by chatGPT

BiCrystal is a Python 3 program that builds incommensurate unit cells of bilayer materials.

Time-reversal Euclidean neural networks based on e3nn

Lectures on QuTiP: slides and ipython notebooks