ORNL / tx2Links
Transformer eXplainability and eXploration
☆19Updated 11 months ago
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
- ISO 4 abbreviation of publication titles in Python☆20Updated last year
- ☆106Updated last year
- complex tensor plugin for pytorch (deprecated)☆47Updated 6 years ago
- matplotlib backend for graph output in unicode terminals using drawille☆80Updated 2 years ago
- Vectorizers for a range of different data types☆103Updated 8 months ago
- ☆66Updated last year
- Test pytorch code with minimal computational overhead☆26Updated 2 years ago
- Reproducible research and reusable acyclic workflows in Python. Execute code on HPC systems as if you executed them on your personal comp…☆18Updated 3 years ago
- ☆60Updated 3 years 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
- Dive into Jax, Flax, XLA and C++☆32Updated 5 years ago
- NoPdb: Non-interactive Python Debugger☆84Updated 3 years ago
- [ICML 2024] SIRFShampoo: Structured inverse- and root-free Shampoo in PyTorch (https://arxiv.org/abs/2402.03496)☆14Updated 11 months ago
- ☆108Updated 2 years ago
- The LUMI AI Guide is designed to assist users in migrating their machine learning applications from smaller-scale computing environments …☆56Updated last month
- The accompanying code for "Simplifying and Understanding State Space Models with Diagonal Linear RNNs" (Ankit Gupta, Harsh Mehta, Jonatha…☆23Updated 2 years ago
- Documentation:☆123Updated 2 years ago
- Quantification of Uncertainty with Adversarial Models☆29Updated 2 years ago
- ☆63Updated last year
- Composable kernels for scikit-learn implemented in JAX.☆44Updated 4 years ago
- ☆14Updated last year
- Tools for the symbolic manipulation of PyMC models, Theano, and TensorFlow graphs.☆63Updated last year
- A snappy + easy + pretty TUI debugger for Python.☆66Updated last week
- Gaussian process time-frequency analysis toy examples☆17Updated 6 years ago
- PyTorch-based library for various kinds of representational-similarity analysis☆24Updated last year
- Pytorch implementation of SuperPolyak subgradient method.☆43Updated 2 years ago
- AdamW optimizer for bfloat16 models in pytorch 🔥.☆37Updated last year
- Bayesian algorithm execution (BAX)☆50Updated 4 years ago