nikitadurasov / torch-ttt
A modular and easy-to-use framework for Test-Time Training (TTT) and Test-Time Adaptation (TTA) in Pytorch, making your networks more generalizable with minimal effort ✨
☆18Updated this week
Alternatives and similar repositories for torch-ttt:
Users that are interested in torch-ttt are comparing it to the libraries listed below
- Stochastic Normalizing Flows☆76Updated 3 years ago
- Flow Annealed Importance Sampling Bootstrap (FAB). ICLR 2023.☆57Updated last year
- Implementation of Action Matching for the Schrödinger equation☆24Updated last year
- Code repository of the paper "Clifford-Steerable Convolutional Neural Networks"☆47Updated 7 months ago
- Official implementation of Metric Flow Matching (NeurIPS 2024)☆35Updated 4 months ago
- Convolutional Differential Operators for Physics-based Deep Learning Study☆24Updated 8 months ago
- Transformers with doubly stochastic attention☆45Updated 2 years ago
- ☆10Updated 2 years ago
- Erwin: A Tree-based Hierarchical Transformer for Large-scale Physical Systems☆42Updated this week
- Implementation of Action Matching☆40Updated last year
- ☆28Updated 7 months ago
- Collecting research materials on neural samplers with diffusion/flow models☆44Updated last month
- Official repository of Implicit Neural Convolutional Kernels for Steerable CNNs, Zhdanov et al.☆28Updated last month
- Official repo for Trumpets: Injective Flows for Inference and Inverse Problems☆13Updated 3 years ago
- Normalizing Flows with a resampled base distribution☆46Updated 2 years ago
- Official code repository of paper Equivariance via Minimal Frame Averaging for More Symmetries and Efficiency.☆17Updated 2 months ago
- Fast, Expressive SE(n) Equivariant Networks through Weight-Sharing in Position-Orientation Space.☆81Updated 10 months ago
- Official code for "Enabling Uncertainty Estimation in Iterative Neural Networks" (ICML 2024)☆16Updated 9 months ago
- ☆17Updated 3 years ago
- Zonal Spherical Harmonics in d Dimensions in TensorFlow, PyTorch and Jax☆28Updated 10 months ago
- Library to make any existing neural network architecture equivariant☆51Updated 5 months ago
- Official implementation of Deep Momentum Schrödinger Bridge☆24Updated last year
- Library for normalizing flows and neural flows.☆24Updated 2 years ago
- Implementation of papers in 101 lines of code.☆18Updated last year
- Code for "Generalised Implicit Neural Representations" (NeurIPS 2022).☆70Updated 2 years ago
- Bare-bones implementations of some generative models in Jax: diffusion, normalizing flows, consistency models, flow matching, (beta)-VAEs…☆128Updated last year
- This repository implements and evaluates convolutional networks on the Möbius strip as toy model instantiations of Coordinate Independent…☆72Updated last year
- ☆115Updated 4 months ago
- ☆18Updated last year
- Use scipy.optimize.minimize as a PyTorch Optimizer.☆72Updated 8 months ago