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:
- A home for audio ML in JAX. Has common features, learnable frontends, pretrained supervised and self-supervised models.☆70Updated 3 years ago
- Test pytorch code with minimal computational overhead☆26Updated 2 years ago
- Quasi-Newton algorithm for joint-diagonalization☆34Updated 3 years ago
- ☆108Updated 3 years ago
- ☆67Updated last year
- Composable kernels for scikit-learn implemented in JAX.☆47Updated 5 years ago
- ☆109Updated last year
- Reproducible research and reusable acyclic workflows in Python. Execute code on HPC systems as if you executed them on your personal comp…☆18Updated 4 years ago
- matplotlib backend for graph output in unicode terminals using drawille☆80Updated 2 years ago
- Bayesian algorithm execution (BAX)☆55Updated 4 years ago
- Vectorizers for a range of different data types☆103Updated 3 months ago
- Library for normalizing flows and neural flows.☆26Updated 3 years ago
- ☆23Updated 4 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
- ☆60Updated 3 years ago
- Gaussian process time-frequency analysis toy examples☆17Updated 6 years ago
- NoPdb: Non-interactive Python Debugger☆84Updated 3 years ago
- Short-time Fourier transform (STFT) for JAX☆15Updated 4 years ago
- AdamW optimizer for bfloat16 models in pytorch 🔥.☆39Updated last year
- Relative Positional Encoding for Transformers with Linear Complexity☆65Updated 3 years ago
- The Recognizing, Exploring, and Articulating Limitations in Machine Learning research tool (REAL ML) is a set of guided activities to hel…☆52Updated 3 years ago
- ISO 4 abbreviation of publication titles in Python☆20Updated 2 years ago
- Visualize tensors in a plain Python REPL using Sparklines☆45Updated 4 years ago
- rho_VAE: an autoregressive parametrization of the VAE encoder☆16Updated 6 years ago
- [ICML 2024] SIRFShampoo: Structured inverse- and root-free Shampoo in PyTorch (https://arxiv.org/abs/2402.03496)☆15Updated last year
- Documentation:☆126Updated 2 years ago
- Efficient Householder Transformation in PyTorch☆69Updated 4 years ago
- Code repository of the paper "Wavelet Networks: Scale-Translation Equivariant Learning From Raw Time-Series, TMLR" https://arxiv.org/abs…☆84Updated 2 years ago
- Generic PyTorch implementation of einsum that supports different semirings☆51Updated 2 months ago
- ☆21Updated 3 years ago