10 projects
jrmt
JAXed Random Matrix Theory
thrmt
Torched Random Matrix Theory
optimuon
A performance-optimized Muon optimizer with foreach support, auto-routing, and composite optimizer patterns.
halea
Unified access to (some) hardware RNGs for machine learning & scientific computing
ebtorch
Collection of PyTorch additions, extensions, utilities, uses and abuses
kuat
Quaternions in PyTorch (part of NVIDIA Kaolin)
foveatorch
Differentiable foveated vision for Deep Learning methods
pip-install-if-missing
Easily install Python packages if they are missing. Thought for Colab-like disposable environments.
fastwonn
Fast, GPU-friendly, differentiable computation of Intrinsic Dimension via Maximum Likelihood, the TwoNN algorithm, and everything in between!
tyxe
BNNs for PyTorch using Pyro
