juliuskunze / jaxnet

Related projects ⓘ

Alternatives and complementary repositories for jaxnet

• harvardnlp / namedtensor
  Named Tensor implementation for Torch
  ☆443Updated 2 years ago
• srush / parallax
  ☆155Updated 4 years ago
• facebookresearch / qmnist
  The QMNIST dataset
  ☆237Updated 2 years ago
• facebookresearch / enorm
  Equi-normalization of Neural Networks
  ☆115Updated 5 years ago
• AlexiaJM / MaximumMarginGANs
  Code for paper: "Support Vector Machines, Wasserstein's distance and gradient-penalty GANs maximize a margin"
  ☆179Updated 4 years ago
• szymonmaszke / torchfunc
  PyTorch functions and utilities to make your life easier
  ☆195Updated 3 years ago
• akosiorek / forge
  Basic experiment framework for tensorflow.
  ☆92Updated 3 years ago
• leonardblier / alrao
  Implementation of "Learning with Random Learning Rates" in PyTorch.
  ☆102Updated 5 years ago
• tensorflow / kfac
  An implementation of KFAC for TensorFlow
  ☆197Updated 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
• pytorch / contrib
  Implementations of ideas from recent papers
  ☆391Updated 3 years ago
• ericjang / maml-jax
  Implementation of Model-Agnostic Meta-Learning (MAML) in Jax
  ☆188Updated 2 years ago
• google-deepmind / jaxline
  ☆152Updated 11 months ago
• jonasrauber / eagerpy
  PyTorch, TensorFlow, JAX and NumPy — all of them natively using the same code
  ☆695Updated last year
• amirgholami / anode
  [IJCAI'19, NeurIPS'19] Anode: Unconditionally Accurate Memory-Efficient Gradients for Neural ODEs
  ☆104Updated 4 years ago
• ofnote / tsalib
  Tensor Shape Annotation Library (numpy, tensorflow, pytorch, ...)
  ☆262Updated 4 years ago
• google-deepmind / jmp
  JMP is a Mixed Precision library for JAX.
  ☆187Updated 6 months ago
• pytorch / nestedtensor
  [Prototype] Tools for the concurrent manipulation of variably sized Tensors.
  ☆253Updated 2 years ago
• mariogeiger / hessian
  hessian in pytorch
  ☆186Updated 4 years ago
• cgarciae / treex
  A Pytree Module system for Deep Learning in JAX
  ☆214Updated last year
• tensorpack / dataflow
  Efficient Data Loading Pipeline in Pure Python
  ☆210Updated 4 years ago
• yaroslavvb / kfac_pytorch
  ☆132Updated 7 years ago
• inferno-pytorch / inferno
  A utility library around PyTorch
  ☆244Updated last year
• AI-DI / Brancher
  A user-centered Python package for differentiable probabilistic inference
  ☆203Updated 4 years ago
• cheind / tf-matplotlib
  Seamlessly integrate matplotlib figures as tensorflow summaries.
  ☆119Updated 5 years ago
• ericjang / nf-jax
  Normalizing Flows in Jax
  ☆105Updated 4 years ago
• Vermeille / Torchelie
  Torchélie is a set of utility functions, layers, losses, models, trainers and other things for PyTorch.
  ☆111Updated 2 weeks ago
• google-deepmind / spectral_inference_networks
  Implementation of Spectral Inference Networks, ICLR 2019
  ☆171Updated 5 years ago
• szymonmaszke / torchdatasets
  PyTorch dataset extended with map, cache etc. (tensorflow.data like)
  ☆328Updated 2 years ago