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
- An implementation of KFAC for TensorFlow☆198Updated 3 years ago
- PyTorch functions and utilities to make your life easier☆195Updated 4 years ago
- Implementation of Model-Agnostic Meta-Learning (MAML) in Jax☆191Updated 2 years ago
- Equi-normalization of Neural Networks☆115Updated 6 years ago
- Implementation of Spectral Inference Networks, ICLR 2019☆172Updated 6 years ago
- Loss Patterns of Neural Networks☆85Updated 3 years ago
- Torchélie is a set of utility functions, layers, losses, models, trainers and other things for PyTorch.☆110Updated 7 months ago
- Basic experiment framework for tensorflow.☆91Updated 4 years ago
- 👩 Pytorch and Jax code for the Madam optimiser.☆51Updated 4 years ago
- Code for paper: "Support Vector Machines, Wasserstein's distance and gradient-penalty GANs maximize a margin"☆178Updated 5 years ago
- DeepOBS: A Deep Learning Optimizer Benchmark Suite☆107Updated last year
- Tensor Shape Annotation Library (numpy, tensorflow, pytorch, ...)☆267Updated 5 years ago
- ☆158Updated last year
- Named Tensor implementation for Torch☆446Updated 3 years ago
- Documentation:☆121Updated 2 years ago
- Normalizing Flows in Jax☆107Updated 4 years ago
- Probabilistic classification in PyTorch/TensorFlow/scikit-learn with Fenchel-Young losses☆186Updated last year
- Code for Neural Arithmetic Units (ICLR) and Measuring Arithmetic Extrapolation Performance (SEDL|NeurIPS)☆146Updated 3 years ago
- [IJCAI'19, NeurIPS'19] Anode: Unconditionally Accurate Memory-Efficient Gradients for Neural ODEs☆107Updated 4 years ago
- hessian in pytorch☆187Updated 4 years ago
- PyTorch, TensorFlow, JAX and NumPy — all of them natively using the same code☆696Updated 2 years ago
- A lightweight library for tensorflow 2.0☆66Updated 5 years ago
- Experiment code for "Randomized Automatic Differentiation"☆67Updated 5 years ago
- A user-centered Python package for differentiable probabilistic inference☆202Updated 4 years ago
- Differentiable bitonic sorting☆141Updated 5 years ago
- Code for: Implicit Competitive Regularization in GANs☆114Updated 3 years ago
- Configure Python functions explicitly and safely☆126Updated 8 months ago
- A probabilistic programming system for simulators and high-performance computing (HPC), based on PyTorch☆389Updated last year
- The QMNIST dataset☆241Updated 2 years ago