Alternatives and similar repositories for jaxnet:

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

• srush / parallax
  ☆153Updated 4 years ago
• yaroslavvb / kfac_pytorch
  ☆132Updated 7 years ago
• facebookresearch / enorm
  Equi-normalization of Neural Networks
  ☆115Updated 5 years ago
• tensorflow / kfac
  An implementation of KFAC for TensorFlow
  ☆198Updated 2 years ago
• akosiorek / forge
  Basic experiment framework for tensorflow.
  ☆92Updated 3 years ago
• google-deepmind / jaxline
  ☆153Updated last year
• Santosh-Gupta / SpeedTorch
  Library for faster pinned CPU <-> GPU transfer in Pytorch
  ☆684Updated 4 years ago
• leonardblier / alrao
  Implementation of "Learning with Random Learning Rates" in PyTorch.
  ☆102Updated 5 years ago
• oval-group / mlogger
  a lightweight and simple logger for Machine Learning
  ☆127Updated 4 years ago
• MagNet-DL / magnet
  Deep Learning Projects that Build Themselves
  ☆360Updated 6 years ago
• google-deepmind / jmp
  JMP is a Mixed Precision library for JAX.
  ☆189Updated last month
• ericjang / maml-jax
  Implementation of Model-Agnostic Meta-Learning (MAML) in Jax
  ☆188Updated 2 years ago
• AlexiaJM / MaximumMarginGANs
  Code for paper: "Support Vector Machines, Wasserstein's distance and gradient-penalty GANs maximize a margin"
  ☆178Updated 4 years ago
• pytorch / contrib
  Implementations of ideas from recent papers
  ☆391Updated 4 years ago
• michaelrzhang / lookahead
  Implementation for the Lookahead Optimizer.
  ☆240Updated 2 years ago
• SymJAX / SymJAX
  Documentation:
  ☆119Updated last year
• amirgholami / anode
  [IJCAI'19, NeurIPS'19] Anode: Unconditionally Accurate Memory-Efficient Gradients for Neural ODEs
  ☆103Updated 4 years ago
• toiaydcdyywlhzvlob / backpack
  This repository is no longer maintained. Check
  ☆81Updated 4 years ago
• ml-research / pau
  Padé Activation Units: End-to-end Learning of Activation Functions in Deep Neural Network
  ☆63Updated 3 years ago
• PetrochukM / HParams
  Configure Python functions explicitly and safely
  ☆126Updated 2 months ago
• inferno-pytorch / inferno
  A utility library around PyTorch
  ☆244Updated 2 years ago
• harvardnlp / namedtensor
  Named Tensor implementation for Torch
  ☆444Updated 2 years ago
• golkar / backdrop
  Implementation and demonstration of backdrop in pytorch. Code and demonstration of GP dataset generator.
  ☆68Updated 6 years ago
• szymonmaszke / torchfunc
  PyTorch functions and utilities to make your life easier
  ☆195Updated 3 years ago
• mariogeiger / hessian
  hessian in pytorch
  ☆186Updated 4 years ago
• fastai / imagenet-fast
  ☆303Updated 3 years ago
• cheind / tf-matplotlib
  Seamlessly integrate matplotlib figures as tensorflow summaries.
  ☆119Updated 6 years ago
• jonasrauber / eagerpy
  PyTorch, TensorFlow, JAX and NumPy — all of them natively using the same code
  ☆695Updated last year
• jotaf98 / overboard
  Pure Python dashboard for monitoring deep learning experiments (like TensorBoard for PyTorch/JAX/etc, without a browser)
  ☆95Updated 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…
  ☆148Updated 7 years ago