ORNL / tx2Links
Transformer eXplainability and eXploration
☆19Updated last year
Alternatives and similar repositories for tx2
Users that are interested in tx2 are comparing it to the libraries listed below
Sorting:
- Quasi-Newton algorithm for joint-diagonalization☆33Updated 3 years ago
- A home for audio ML in JAX. Has common features, learnable frontends, pretrained supervised and self-supervised models.☆69Updated 3 years ago
- Gaussian process time-frequency analysis toy examples☆17Updated 6 years ago
- complex tensor plugin for pytorch (deprecated)☆47Updated 6 years ago
- Composable kernels for scikit-learn implemented in JAX.☆45Updated 5 years ago
- Test pytorch code with minimal computational overhead☆26Updated 2 years ago
- ☆67Updated last year
- [ICML 2024] SIRFShampoo: Structured inverse- and root-free Shampoo in PyTorch (https://arxiv.org/abs/2402.03496)☆14Updated last year
- Dive into Jax, Flax, XLA and C++☆32Updated 5 years ago
- Code repository of the paper "Wavelet Networks: Scale-Translation Equivariant Learning From Raw Time-Series, TMLR" https://arxiv.org/abs…☆84Updated last year
- Library for normalizing flows and neural flows.☆26Updated 3 years ago
- Pytorch implementation of SuperPolyak subgradient method.☆43Updated 3 years ago
- Bibtex for various Python science and machine learning software☆33Updated 3 years ago
- ☆107Updated last year
- PyMC3-like Interface for Pyro☆38Updated 6 years ago
- The LUMI AI Guide is designed to assist users in migrating their machine learning applications from smaller-scale computing environments …☆61Updated last month
- ☆23Updated 4 years ago
- You should use PySR to find scaling laws. Here's an example.☆33Updated 2 years ago
- Bayesian algorithm execution (BAX)☆55Updated 4 years ago
- Documentation:☆125Updated 2 years ago
- Visualize, create, and operate on pytrees in the most intuitive way possible.☆45Updated 11 months ago
- Bayesian entropy estimation in Python - via the Nemenman-Schafee-Bialek algorithm☆50Updated 5 years ago
- The accompanying code for "Simplifying and Understanding State Space Models with Diagonal Linear RNNs" (Ankit Gupta, Harsh Mehta, Jonatha…☆23Updated 3 years ago
- Python library for analyzing the internal structure of deep neural networks.☆19Updated last month
- ☆21Updated 3 years ago
- Bayesian histograms for estimation of binary rare event rates, with fully automated bin pruning☆29Updated 4 years ago
- ☆60Updated 3 years ago
- Python implementation of projection losses.☆27Updated 6 years ago
- ISO 4 abbreviation of publication titles in Python☆20Updated 2 years ago
- Short-time Fourier transform (STFT) for JAX☆15Updated 4 years ago