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
• szymonmaszke / torchfunc
  PyTorch functions and utilities to make your life easier
  ☆195Updated 4 years ago
• jonasrauber / eagerpy
  PyTorch, TensorFlow, JAX and NumPy — all of them natively using the same code
  ☆696Updated 2 years ago
• tensorflow / kfac
  An implementation of KFAC for TensorFlow
  ☆198Updated 3 years ago
• SymJAX / SymJAX
  Documentation:
  ☆120Updated 2 years ago
• AlexiaJM / MaximumMarginGANs
  Code for paper: "Support Vector Machines, Wasserstein's distance and gradient-penalty GANs maximize a margin"
  ☆178Updated 5 years ago
• facebookresearch / enorm
  Equi-normalization of Neural Networks
  ☆115Updated 6 years ago
• yaroslavvb / kfac_pytorch
  ☆133Updated 7 years ago
• AI-DI / Brancher
  A user-centered Python package for differentiable probabilistic inference
  ☆202Updated 4 years ago
• jxbz / madam
  👩 Pytorch and Jax code for the Madam optimiser.
  ☆51Updated 4 years ago
• akosiorek / forge
  Basic experiment framework for tensorflow.
  ☆92Updated 4 years ago
• Vermeille / Torchelie
  Torchélie is a set of utility functions, layers, losses, models, trainers and other things for PyTorch.
  ☆110Updated 6 months ago
• leonardblier / alrao
  Implementation of "Learning with Random Learning Rates" in PyTorch.
  ☆102Updated 5 years ago
• ofnote / tsalib
  Tensor Shape Annotation Library (numpy, tensorflow, pytorch, ...)
  ☆266Updated 5 years ago
• amirgholami / anode
  [IJCAI'19, NeurIPS'19] Anode: Unconditionally Accurate Memory-Efficient Gradients for Neural ODEs
  ☆104Updated 4 years ago
• cgarciae / treex
  A Pytree Module system for Deep Learning in JAX
  ☆214Updated 2 years ago
• michaelrzhang / lookahead
  Implementation for the Lookahead Optimizer.
  ☆240Updated 3 years ago
• google-deepmind / mc_gradients
  ☆168Updated 10 months ago
• ericjang / maml-jax
  Implementation of Model-Agnostic Meta-Learning (MAML) in Jax
  ☆190Updated 2 years ago
• harvardnlp / namedtensor
  Named Tensor implementation for Torch
  ☆445Updated 2 years ago
• toiaydcdyywlhzvlob / backpack
  This repository is no longer maintained. Check
  ☆81Updated 5 years ago
• google-deepmind / jaxline
  ☆157Updated last year
• inferno-pytorch / inferno
  A utility library around PyTorch
  ☆244Updated 2 years ago
• pclucas14 / pytorch-glow
  Pytorch Implementation of OpenAI's GLOW
  ☆94Updated 4 years ago
• mblondel / fenchel-young-losses
  Probabilistic classification in PyTorch/TensorFlow/scikit-learn with Fenchel-Young losses
  ☆185Updated last year
• ericjang / nf-jax
  Normalizing Flows in Jax
  ☆108Updated 4 years ago
• pytorch / contrib
  Implementations of ideas from recent papers
  ☆392Updated 4 years ago
• pyprob / pyprob
  A probabilistic programming system for simulators and high-performance computing (HPC), based on PyTorch
  ☆391Updated last year
• SirRob1997 / Crowded-Valley---Results
  This repository contains the results for the paper: "Descending through a Crowded Valley - Benchmarking Deep Learning Optimizers"
  ☆180Updated 3 years ago
• google-deepmind / spectral_inference_networks
  Implementation of Spectral Inference Networks, ICLR 2019
  ☆172Updated 6 years ago