deterministic-algorithms-lab / Jax-Journey
A pathway and collection of resources to learning Jax from beginning to advance.
☆10Updated 4 years ago
Alternatives and similar repositories for Jax-Journey
Users that are interested in Jax-Journey are comparing it to the libraries listed below
Sorting:
- Simple, extensible implementations of some meta-learning algorithms in Jax☆10Updated 4 years ago
- Local Attention - Flax module for Jax☆21Updated 3 years ago
- JAX implementation of Graph Attention Networks☆13Updated 3 years ago
- A framework for implementing equivariant DL☆10Updated 3 years ago
- This repository hosts the code to port NumPy model weights of BiT-ResNets to TensorFlow SavedModel format.☆14Updated 3 years ago
- JAX implementation of Learning to learn by gradient descent by gradient descent☆27Updated 7 months ago
- AdaCat☆49Updated 2 years ago
- Clockwork VAEs in JAX/Flax☆32Updated 3 years ago
- Flax (JAX) implementation of Progressive Growing of GANs for Improved Quality, Stability, and Variation☆12Updated 3 years ago
- Probabilistic Solution of Differential Equations☆13Updated 2 years ago
- Official code for Long Expressive Memory (ICLR 2022, Spotlight)☆69Updated 3 years ago
- ☆31Updated last month
- A JAX implementation of stochastic addition.☆14Updated 2 years ago
- A stateful pytree library for training neural networks.☆22Updated 2 years ago
- Automatically generate simple meta-learning tasks from a very large space☆15Updated last year
- ☆15Updated 2 years ago
- Easily serialize dataclasses to and from tensors (PyTorch, NumPy)☆18Updated 4 years ago
- Fine-grained, dynamic control of neural network topology in JAX.☆21Updated last year
- ☆33Updated 2 years ago
- Model Agnostic Meta Learning (MAML) implemented in Flax, the neural network library for JAX.☆19Updated 4 years ago
- Usable implementation of Emerging Symbol Binding Network (ESBN), in Pytorch☆25Updated 4 years ago
- Image augmentation library for Jax☆39Updated last year
- Supplementary code for the paper "Meta-Solver for Neural Ordinary Differential Equations" https://arxiv.org/abs/2103.08561☆25Updated 4 years ago
- A simple Transformer where the softmax has been replaced with normalization☆20Updated 4 years ago
- ☆26Updated 2 years ago
- A selection of neural network models ported from torchvision for JAX & Flax.☆44Updated 4 years ago
- notebooks of cool EBM visualizations☆15Updated 4 years ago
- Test pytorch code with minimal computational overhead☆25Updated last year
- ☆15Updated 4 years ago
- flexible meta-learning in jax☆13Updated last year