Alternatives and similar repositories for jaxnet:

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

• tensorflow / kfac
  An implementation of KFAC for TensorFlow
  ☆198Updated 3 years ago
• srush / parallax
  ☆153Updated 4 years ago
• facebookresearch / enorm
  Equi-normalization of Neural Networks
  ☆115Updated 5 years ago
• Vermeille / Torchelie
  Torchélie is a set of utility functions, layers, losses, models, trainers and other things for PyTorch.
  ☆110Updated 3 months ago
• akosiorek / forge
  Basic experiment framework for tensorflow.
  ☆92Updated 3 years ago
• pytorch / contrib
  Implementations of ideas from recent papers
  ☆392Updated 4 years ago
• szymonmaszke / torchfunc
  PyTorch functions and utilities to make your life easier
  ☆195Updated 4 years ago
• AndreasMadsen / stable-nalu
  Code for Neural Arithmetic Units (ICLR) and Measuring Arithmetic Extrapolation Performance (SEDL|NeurIPS)
  ☆147Updated 3 years ago
• AlexiaJM / MaximumMarginGANs
  Code for paper: "Support Vector Machines, Wasserstein's distance and gradient-penalty GANs maximize a margin"
  ☆178Updated 5 years ago
• devzhk / Implicit-Competitive-Regularization
  Code for: Implicit Competitive Regularization in GANs
  ☆114Updated 3 years ago
• yaroslavvb / kfac_pytorch
  ☆133Updated 7 years ago
• leonardblier / alrao
  Implementation of "Learning with Random Learning Rates" in PyTorch.
  ☆102Updated 5 years ago
• ericjang / maml-jax
  Implementation of Model-Agnostic Meta-Learning (MAML) in Jax
  ☆189Updated 2 years ago
• google-deepmind / jaxline
  ☆154Updated last year
• SymJAX / SymJAX
  Documentation:
  ☆119Updated last year
• harvardnlp / namedtensor
  Named Tensor implementation for Torch
  ☆444Updated 2 years ago
• tensorpack / dataflow
  Efficient Data Loading Pipeline in Pure Python
  ☆211Updated 4 years ago
• Santosh-Gupta / SpeedTorch
  Library for faster pinned CPU <-> GPU transfer in Pytorch
  ☆685Updated 5 years ago
• pytorch / nestedtensor
  [Prototype] Tools for the concurrent manipulation of variably sized Tensors.
  ☆251Updated 2 years ago
• eladhoffer / bigBatch
  Code used to generate the results appearing in "Train longer, generalize better: closing the generalization gap in large batch training o…
  ☆149Updated 7 years ago
• cheind / tf-matplotlib
  Seamlessly integrate matplotlib figures as tensorflow summaries.
  ☆120Updated 6 years ago
• inferno-pytorch / inferno
  A utility library around PyTorch
  ☆244Updated 2 years ago
• mblondel / fenchel-young-losses
  Probabilistic classification in PyTorch/TensorFlow/scikit-learn with Fenchel-Young losses
  ☆184Updated last year
• taki0112 / RAdam-Tensorflow
  Simple Tensorflow implementation of "On The Variance Of The Adaptive Learning Rate And Beyond"
  ☆97Updated 5 years ago
• szymonmaszke / torchdatasets
  PyTorch dataset extended with map, cache etc. (tensorflow.data like)
  ☆329Updated 2 years ago
• jonasrauber / eagerpy
  PyTorch, TensorFlow, JAX and NumPy — all of them natively using the same code
  ☆696Updated 2 years ago
• CyberZHG / keras-adabound
  AdaBound optimizer in Keras
  ☆56Updated 4 years ago
• AI-DI / Brancher
  A user-centered Python package for differentiable probabilistic inference
  ☆202Updated 4 years ago
• negrinho / deep_architect
  A general, modular, and programmable architecture search framework
  ☆123Updated 2 years ago
• google-deepmind / jmp
  JMP is a Mixed Precision library for JAX.
  ☆194Updated 2 months ago