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
- Basic experiment framework for tensorflow.☆91Updated 4 years ago
- PyTorch functions and utilities to make your life easier☆194Updated 4 years ago
- Implementation of Model-Agnostic Meta-Learning (MAML) in Jax☆190Updated 3 years ago
- Equi-normalization of Neural Networks☆115Updated 6 years ago
- Loss Patterns of Neural Networks☆85Updated 4 years ago
- Normalizing Flows in Jax☆107Updated 5 years ago
- An implementation of KFAC for TensorFlow☆198Updated 3 years ago
- Tensor Shape Annotation Library (numpy, tensorflow, pytorch, ...)☆266Updated 5 years ago
- DeepOBS: A Deep Learning Optimizer Benchmark Suite☆108Updated last year
- Implementation of Spectral Inference Networks, ICLR 2019☆172Updated 6 years ago
- [IJCAI'19, NeurIPS'19] Anode: Unconditionally Accurate Memory-Efficient Gradients for Neural ODEs☆106Updated 4 years ago
- Code for Neural Arithmetic Units (ICLR) and Measuring Arithmetic Extrapolation Performance (SEDL|NeurIPS)☆146Updated 4 years ago
- Named Tensor implementation for Torch☆446Updated 3 years ago
- 👩 Pytorch and Jax code for the Madam optimiser.☆52Updated 4 years ago
- Torchélie is a set of utility functions, layers, losses, models, trainers and other things for PyTorch.☆110Updated last month
- The QMNIST dataset☆241Updated 3 years ago
- ☆183Updated last year
- ☆158Updated last year
- Implements stochastic line search☆118Updated 2 years ago
- Code for paper: "Support Vector Machines, Wasserstein's distance and gradient-penalty GANs maximize a margin"☆179Updated 5 years ago
- Mixture Density Networks (Bishop, 1994) tutorial in JAX☆60Updated 5 years ago
- ☆133Updated 7 years ago
- ☆170Updated last year
- Code for: Implicit Competitive Regularization in GANs☆116Updated 3 years ago
- Documentation:☆123Updated 2 years ago
- Sample implementation of Neural Ordinary Differential Equations☆262Updated 6 years ago
- Experiment code for "Randomized Automatic Differentiation"☆67Updated 5 years ago
- A general, modular, and programmable architecture search framework☆124Updated 2 years ago
- Probabilistic classification in PyTorch/TensorFlow/scikit-learn with Fenchel-Young losses☆190Updated 2 years ago