francois-rozet / torchist

Related projects ⓘ

Alternatives and complementary repositories for torchist

• toshas / torch-householder
  Efficient Householder Transformation in PyTorch
  ☆60Updated 3 years ago
• sbarratt / torch_interpolations
  Interpolation routines in Pytorch.
  ☆109Updated 3 years ago
• rolandgvc / flaxvision
  A selection of neural network models ported from torchvision for JAX & Flax.
  ☆44Updated 3 years ago
• f-dangel / vivit
  [TMLR 2022] Curvature access through the generalized Gauss-Newton's low-rank structure: Eigenvalues, eigenvectors, directional derivative…
  ☆17Updated last year
• unixpickle / vae-textures
  Texture mapping with variational auto-encoders
  ☆40Updated 3 years ago
• google-deepmind / einshape
  ☆98Updated 4 months ago
• CW-Huang / CP-Flow
  Convex potential flows
  ☆78Updated 2 years ago
• AlexanderMath / fasth
  Code for the article "What if Neural Networks had SVDs?", to be presented as a spotlight paper at NeurIPS 2020.
  ☆69Updated 3 months ago
• gbaydin / difftorch
  A differentiation API for PyTorch
  ☆30Updated 4 years ago
• crowsonkb / mdmm
  The Modified Differential Multiplier Method (MDMM) for PyTorch
  ☆51Updated 3 years ago
• titu1994 / tf_fourier_features
  Tensorflow 2.0 implementation of Fourier Feature Mapping Networks.
  ☆42Updated 4 years ago
• cgarciae / treeo
  A small library for creating and manipulating custom JAX Pytree classes
  ☆58Updated last year
• aliutkus / torchsearchsorted
  Pytorch Custom CUDA kernel for searchsorted
  ☆135Updated last year
• arogozhnikov / eindex
  Multidimensional indexing for tensors
  ☆112Updated last year
• f-dangel / sirfshampoo
  [ICML 2024] SIRFShampoo: Structured inverse- and root-free Shampoo in PyTorch (https://arxiv.org/abs/2402.03496)
  ☆13Updated this week
• patrick-kidger / FasterNeuralDiffEq
  Code for "'Hey, that's not an ODE:' Faster ODE Adjoints via Seminorms" (ICML 2021)
  ☆86Updated 2 years ago
• SymJAX / SymJAX
  Documentation:
  ☆120Updated last year
• n2cholas / jax-resnet
  Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).
  ☆104Updated 2 years ago
• juntang-zhuang / torch_ACA
  repo for paper: Adaptive Checkpoint Adjoint (ACA) method for gradient estimation in neural ODE
  ☆55Updated 3 years ago
• SarthakYadav / audax
  A home for audio ML in JAX. Has common features, learnable frontends, pretrained supervised and self-supervised models.
  ☆62Updated 2 years ago
• jxbz / nero
  👑 Pytorch code for the Nero optimiser.
  ☆20Updated 2 years ago
• ericjang / nf-jax
  Normalizing Flows in Jax
  ☆105Updated 4 years ago
• tk-rusch / LEM
  Official code for Long Expressive Memory (ICLR 2022, Spotlight)
  ☆69Updated 2 years ago
• akandykeller / SelfNormalizingFlows
  ☆67Updated 2 years ago
• Information-Fusion-Lab-Umass / NuX
  Normalizing Flows using JAX
  ☆82Updated 11 months ago
• gisilvs / AEF
  ☆33Updated last year
• ltatzel / PyTorchHessianFree
  PyTorch implementation of the Hessian-free optimizer
  ☆30Updated 4 months ago
• lixilinx / psgd_torch
  Pytorch implementation of preconditioned stochastic gradient descent (affine group preconditioner, low-rank approximation preconditioner …
  ☆110Updated 3 weeks ago
• f-dangel / einconv
  Convolutions and more as einsum for PyTorch
  ☆13Updated 5 months ago
• shreyansh26 / ML-Optimizers-JAX
  Toy implementations of some popular ML optimizers using Python/JAX
  ☆43Updated 3 years ago