ORNL / tx2
Transformer eXplainability and eXploration
☆19Updated 6 months ago
Alternatives and similar repositories for tx2
Users that are interested in tx2 are comparing it to the libraries listed below
Sorting:
- You should use PySR to find scaling laws. Here's an example.☆33Updated last year
- Dive into Jax, Flax, XLA and C++☆31Updated 5 years ago
- Vectorizers for a range of different data types☆101Updated 3 months ago
- Graph Learning with JAX☆14Updated 2 years ago
- Python library for analyzing the internal structure of deep neural networks.☆18Updated last year
- ISO 4 abbreviation of publication titles in Python☆20Updated last year
- ☆103Updated 10 months ago
- Pytorch implementation of SuperPolyak subgradient method.☆43Updated 2 years ago
- Fully and Partially Bayesian Neural Nets☆68Updated 2 weeks ago
- An experiment workflow and organization tool.☆18Updated 7 months ago
- Pure C implementation of e3nn☆18Updated 2 months ago
- funsies is a lightweight workflow engine 🔧☆41Updated 3 years ago
- Test pytorch code with minimal computational overhead☆25Updated last year
- ☆50Updated 2 years ago
- ☆10Updated 2 years ago
- SciML Benchmarking Suite for AI for Science☆40Updated 9 months ago
- ☆60Updated 3 years ago
- ☆55Updated last year
- Composable kernels for scikit-learn implemented in JAX.☆43Updated 4 years ago
- GPU-Parallelized UMAP☆52Updated 5 years ago
- Distance-based Analysis of DAta-manifolds in python☆127Updated last month
- ☆61Updated last year
- ☆14Updated last year
- 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
- Deep Coarse-grained Potentials via Relative Entropy Minimization☆14Updated 2 years ago
- Library for normalizing flows and neural flows.☆24Updated 2 years ago
- Stencil computations in JAX☆71Updated last year
- Implementation of approximate free-energy minimization in PyTorch☆19Updated 3 years ago
- Steerable E(3) GNN in jax☆24Updated last year
- ☆33Updated last year