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:
- PyTorch functions and utilities to make your life easier☆195Updated 4 years ago
- Equi-normalization of Neural Networks☆115Updated 6 years ago
- ☆153Updated 5 years ago
- Code for paper: "Support Vector Machines, Wasserstein's distance and gradient-penalty GANs maximize a margin"☆178Updated 5 years ago
- Basic experiment framework for tensorflow.☆92Updated 3 years ago
- An implementation of KFAC for TensorFlow☆198Updated 3 years ago
- PyTorch, TensorFlow, JAX and NumPy — all of them natively using the same code☆698Updated 2 years ago
- hessian in pytorch☆187Updated 4 years ago
- ☆133Updated 7 years ago
- Implementation for the Lookahead Optimizer.☆240Updated 3 years ago
- Torchélie is a set of utility functions, layers, losses, models, trainers and other things for PyTorch.☆110Updated 5 months ago
- Pytorch Implementation of OpenAI's GLOW☆94Updated 4 years ago
- A user-centered Python package for differentiable probabilistic inference☆202Updated 4 years ago
- Implementations of ideas from recent papers☆393Updated 4 years ago
- Tensor Shape Annotation Library (numpy, tensorflow, pytorch, ...)☆266Updated 5 years ago
- Implementation of Model-Agnostic Meta-Learning (MAML) in Jax☆189Updated 2 years ago
- The QMNIST dataset☆239Updated 2 years ago
- Normalizing Flows in Jax☆108Updated 4 years ago
- Named Tensor implementation for Torch☆444Updated 2 years ago
- Loss Patterns of Neural Networks☆85Updated 3 years ago
- Simple Tensorflow implementation of "On The Variance Of The Adaptive Learning Rate And Beyond"☆97Updated 5 years ago
- Robust Bi-Tempered Logistic Loss Based on Bregman Divergences. https://arxiv.org/pdf/1906.03361.pdf☆148Updated 3 years ago
- Documentation:☆120Updated 2 years ago
- ☆157Updated last year
- Code for: Implicit Competitive Regularization in GANs☆113Updated 3 years ago
- Code for Neural Arithmetic Units (ICLR) and Measuring Arithmetic Extrapolation Performance (SEDL|NeurIPS)☆146Updated 3 years ago
- Seamlessly integrate matplotlib figures as tensorflow summaries.☆120Updated 6 years ago
- A Pytree Module system for Deep Learning in JAX☆214Updated 2 years ago
- Implementation of "Learning with Random Learning Rates" in PyTorch.☆102Updated 5 years ago
- Padé Activation Units: End-to-end Learning of Activation Functions in Deep Neural Network☆64Updated 4 years ago