Alternatives and similar repositories for jaxnet

Users that are interested in jaxnet are comparing it to the libraries listed below

• srush / parallax
  ☆153Updated 5 years ago
• akosiorek / forge
  Basic experiment framework for tensorflow.
  ☆91Updated 4 years ago
• szymonmaszke / torchfunc
  PyTorch functions and utilities to make your life easier
  ☆194Updated 4 years ago
• ericjang / nf-jax
  Normalizing Flows in Jax
  ☆107Updated 5 years ago
• facebookresearch / qmnist
  The QMNIST dataset
  ☆241Updated 3 years ago
• Vermeille / Torchelie
  Torchélie is a set of utility functions, layers, losses, models, trainers and other things for PyTorch.
  ☆110Updated last week
• tensorflow / kfac
  An implementation of KFAC for TensorFlow
  ☆199Updated 3 years ago
• google-deepmind / spectral_inference_networks
  Implementation of Spectral Inference Networks, ICLR 2019
  ☆172Updated 6 years ago
• harvardnlp / namedtensor
  Named Tensor implementation for Torch
  ☆447Updated 3 years ago
• mariogeiger / hessian
  hessian in pytorch
  ☆187Updated 5 years ago
• amirgholami / anode
  [IJCAI'19, NeurIPS'19] Anode: Unconditionally Accurate Memory-Efficient Gradients for Neural ODEs
  ☆109Updated 5 years ago
• AlexiaJM / MaximumMarginGANs
  Code for paper: "Support Vector Machines, Wasserstein's distance and gradient-penalty GANs maximize a margin"
  ☆179Updated 5 years ago
• SymJAX / SymJAX
  Documentation:
  ☆125Updated 2 years ago
• mblondel / fenchel-young-losses
  Probabilistic classification in PyTorch/TensorFlow/scikit-learn with Fenchel-Young losses
  ☆193Updated 2 years ago
• facebookresearch / enorm
  Equi-normalization of Neural Networks
  ☆116Updated 6 years ago
• google-deepmind / mc_gradients
  ☆172Updated last year
• devzhk / Implicit-Competitive-Regularization
  Code for: Implicit Competitive Regularization in GANs
  ☆116Updated 3 years ago
• google-deepmind / jaxline
  ☆161Updated 2 years ago
• PetrochukM / HParams
  Configure Python functions explicitly and safely
  ☆128Updated last year
• universome / loss-patterns
  Loss Patterns of Neural Networks
  ☆86Updated 4 years ago
• jxbz / madam
  👩 Pytorch and Jax code for the Madam optimiser.
  ☆53Updated 4 years ago
• ericjang / maml-jax
  Implementation of Model-Agnostic Meta-Learning (MAML) in Jax
  ☆191Updated 3 years ago
• ofnote / tsalib
  Tensor Shape Annotation Library (numpy, tensorflow, pytorch, ...)
  ☆266Updated 5 years ago
• IssamLaradji / sls
  Implements stochastic line search
  ☆118Updated 2 years ago
• yaroslavvb / kfac_pytorch
  ☆134Updated 8 years ago
• hardmaru / mdn_jax_tutorial
  Mixture Density Networks (Bishop, 1994) tutorial in JAX
  ☆61Updated 5 years ago
• AndreasMadsen / stable-nalu
  Code for Neural Arithmetic Units (ICLR) and Measuring Arithmetic Extrapolation Performance (SEDL|NeurIPS)
  ☆147Updated 4 years ago
• RuiShu / tensorsketch
  A lightweight library for tensorflow 2.0
  ☆65Updated 6 years ago
• google / bi-tempered-loss
  Robust Bi-Tempered Logistic Loss Based on Bregman Divergences. https://arxiv.org/pdf/1906.03361.pdf
  ☆147Updated 4 years ago
• conormdurkan / autoregressive-energy-machines
  Autoregressive Energy Machines
  ☆78Updated 3 years ago