mhlr / awesome-jax
List of awesome JAX resources
☆12Updated last year
Related projects: ⓘ
- Code for our ICLR Trustworthy ML 2020 workshop paper "Improved Image Wasserstein Attacks and Defenses"☆13Updated 4 years ago
- ☆21Updated this week
- High-performance tokenized language data-loader for Python C++ extension☆12Updated last month
- Simplifying parsing of large jsonline files in NLP Workflows☆12Updated 2 years ago
- Companion code for a tutorial on using Hydra.☆27Updated 3 years ago
- LaTeX source code for the slides☆21Updated 3 years ago
- notebooks of cool EBM visualizations☆16Updated 3 years ago
- JAX implementation of Graph Attention Networks☆12Updated 2 years ago
- A runtime shape checker and auto-annotator for tensor programs (pronounced "stanley")☆39Updated 4 years ago
- ☆13Updated last year
- Official repository for our ICLR 2021 paper Evaluating the Disentanglement of Deep Generative Models with Manifold Topology☆36Updated 3 years ago
- Massively Parallel and Asynchronous Architecture for Logic-based AI☆41Updated last year
- Notebooks for IPAM Tutorial, March 15 2019☆24Updated 5 years ago
- An easy way to start a python programming environment using GitHub Codespaces.☆15Updated 4 years ago
- Usable implementation of Emerging Symbol Binding Network (ESBN), in Pytorch☆23Updated 3 years ago
- Implements EvoNorms B0 and S0 as proposed in Evolving Normalization-Activation Layers.☆11Updated 4 years ago
- 👑 Pytorch code for the Nero optimiser.☆20Updated last year
- A framework for implementing equivariant DL☆10Updated 3 years ago
- An attempt to merge ESBN with Transformers, to endow Transformers with the ability to emergently bind symbols☆14Updated 3 years ago
- ☆12Updated 3 years ago
- ☆31Updated 2 years ago
- A collection of utilities for writing labeling functions, transformation functions, and slicing functions.☆20Updated 4 years ago
- Isomorphisms of quiver representations applied to neural networks.☆15Updated last month
- Easily serialize dataclasses to and from tensors (PyTorch, NumPy)☆18Updated 3 years ago
- You should use PySR to find scaling laws. Here's an example.☆30Updated 11 months ago
- ☆10Updated this week
- Deep Critical Learning. Implementation of ProSelfLC, IMAE, DM, etc.☆31Updated last year
- ☆13Updated this week
- Jupyter Notebook corresponding to 'Going with the Flow: An Introduction to Normalizing Flows'☆25Updated 3 years ago
- Quasi-Newton Algorithm for Stochastic Optimization☆10Updated 2 years ago