ORNL / tx2Links
Transformer eXplainability and eXploration
☆19Updated 8 months ago
Alternatives and similar repositories for tx2
Users that are interested in tx2 are comparing it to the libraries listed below
Sorting:
- Fully and Partially Bayesian Neural Nets☆74Updated 2 months ago
- A home for audio ML in JAX. Has common features, learnable frontends, pretrained supervised and self-supervised models.☆68Updated 2 years ago
- Distance-based Analysis of DAta-manifolds in python☆131Updated 2 weeks ago
- The LUMI AI Guide is designed to assist users in migrating their machine learning applications from smaller-scale computing environments …☆50Updated 3 weeks ago
- Library for normalizing flows and neural flows.☆24Updated 3 years ago
- Python library for analyzing the internal structure of deep neural networks.☆18Updated last month
- You should use PySR to find scaling laws. Here's an example.☆33Updated last year
- Test pytorch code with minimal computational overhead☆25Updated 2 years ago
- ISO 4 abbreviation of publication titles in Python☆20Updated last year
- FLOPS counter for all your GPU benchmarking needs☆13Updated 11 months ago
- Bibtex for various Python science and machine learning software☆33Updated 2 years ago
- Bayesian algorithm execution (BAX)☆49Updated 3 years ago
- An implementation of various tensor-based decomposition for NN & RNN parameters☆18Updated 7 years ago
- GroupMap: beyond mean and variance matching for deep learning☆10Updated 2 years ago
- ☆66Updated 10 months ago
- The Recognizing, Exploring, and Articulating Limitations in Machine Learning research tool (REAL ML) is a set of guided activities to hel…☆51Updated 3 years ago
- ☆62Updated last year
- [ICML 2024] SIRFShampoo: Structured inverse- and root-free Shampoo in PyTorch (https://arxiv.org/abs/2402.03496)☆14Updated 8 months ago
- Dive into Jax, Flax, XLA and C++☆31Updated 5 years ago
- Short-time Fourier transform (STFT) for JAX☆14Updated 3 years ago
- Zonal Spherical Harmonics in d Dimensions in TensorFlow, PyTorch and Jax☆33Updated last year
- ☆20Updated 2 years ago
- ☆58Updated last year
- Efficient Householder Transformation in PyTorch☆66Updated 4 years ago
- Create Numpy .npy files by appending on the growth axis☆63Updated 3 weeks ago
- SciML Benchmarking Suite for AI for Science☆42Updated 11 months ago
- Code repository of the paper "Wavelet Networks: Scale-Translation Equivariant Learning From Raw Time-Series, TMLR" https://arxiv.org/abs…☆82Updated last year
- ☆51Updated 2 years ago
- ☆121Updated 7 months ago
- ☆16Updated 4 years ago