juliuskunze / jaxnetLinks
Concise deep learning for JAX
☆183Updated 4 years ago
Alternatives and similar repositories for jaxnet
Users that are interested in jaxnet are comparing it to the libraries listed below
Sorting:
- ☆153Updated 5 years ago
- PyTorch functions and utilities to make your life easier☆195Updated 4 years ago
- PyTorch, TensorFlow, JAX and NumPy — all of them natively using the same code☆696Updated 2 years ago
- An implementation of KFAC for TensorFlow☆198Updated 3 years ago
- Documentation:☆120Updated 2 years ago
- Code for paper: "Support Vector Machines, Wasserstein's distance and gradient-penalty GANs maximize a margin"☆178Updated 5 years ago
- Equi-normalization of Neural Networks☆115Updated 6 years ago
- ☆133Updated 7 years ago
- A user-centered Python package for differentiable probabilistic inference☆202Updated 4 years ago
- 👩 Pytorch and Jax code for the Madam optimiser.☆51Updated 4 years ago
- Basic experiment framework for tensorflow.☆92Updated 4 years ago
- Torchélie is a set of utility functions, layers, losses, models, trainers and other things for PyTorch.☆110Updated 6 months ago
- Implementation of "Learning with Random Learning Rates" in PyTorch.☆102Updated 5 years ago
- Tensor Shape Annotation Library (numpy, tensorflow, pytorch, ...)☆266Updated 5 years ago
- [IJCAI'19, NeurIPS'19] Anode: Unconditionally Accurate Memory-Efficient Gradients for Neural ODEs☆104Updated 4 years ago
- A Pytree Module system for Deep Learning in JAX☆214Updated 2 years ago
- Implementation for the Lookahead Optimizer.☆240Updated 3 years ago
- ☆168Updated 10 months ago
- Implementation of Model-Agnostic Meta-Learning (MAML) in Jax☆190Updated 2 years ago
- Named Tensor implementation for Torch☆445Updated 2 years ago
- This repository is no longer maintained. Check☆81Updated 5 years ago
- ☆157Updated last year
- A utility library around PyTorch☆244Updated 2 years ago
- Pytorch Implementation of OpenAI's GLOW☆94Updated 4 years ago
- Probabilistic classification in PyTorch/TensorFlow/scikit-learn with Fenchel-Young losses☆185Updated last year
- Normalizing Flows in Jax☆108Updated 4 years ago
- Implementations of ideas from recent papers☆392Updated 4 years ago
- A probabilistic programming system for simulators and high-performance computing (HPC), based on PyTorch☆391Updated last year
- This repository contains the results for the paper: "Descending through a Crowded Valley - Benchmarking Deep Learning Optimizers"☆180Updated 3 years ago
- Implementation of Spectral Inference Networks, ICLR 2019☆172Updated 6 years ago