francois-rozet / torchist

Related projects ⓘ

Alternatives and complementary repositories for torchist

• toshas / torch-householder
  Efficient Householder Transformation in PyTorch
  ☆62Updated 3 years ago
• google-deepmind / einshape
  ☆99Updated 4 months ago
• sbarratt / torch_interpolations
  Interpolation routines in Pytorch.
  ☆110Updated 3 years ago
• brentyi / jax_dataclasses
  Pytrees + dataclasses ❤️
  ☆64Updated 4 months 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
  ☆57Updated last year
• SymJAX / SymJAX
  Documentation:
  ☆120Updated last year
• rolandgvc / flaxvision
  A selection of neural network models ported from torchvision for JAX & Flax.
  ☆44Updated 3 years ago
• CW-Huang / CP-Flow
  Convex potential flows
  ☆78Updated 3 years ago
• samuela / torch2jax
  Run PyTorch in JAX. 🤝
  ☆201Updated this week
• crowsonkb / mdmm
  The Modified Differential Multiplier Method (MDMM) for PyTorch
  ☆52Updated 3 years ago
• unixpickle / vae-textures
  Texture mapping with variational auto-encoders
  ☆40Updated 3 years ago
• n2cholas / jax-resnet
  Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).
  ☆104Updated 2 years ago
• COR-OPT / SuperPolyak.py
  Pytorch implementation of SuperPolyak subgradient method.
  ☆43Updated 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
• Information-Fusion-Lab-Umass / NuX
  Normalizing Flows using JAX
  ☆82Updated 11 months ago
• ericjang / nf-jax
  Normalizing Flows in Jax
  ☆105Updated 4 years ago
• cornellius-gp / linear_operator.old
  A LinearOperator implementation for PyTorch
  ☆18Updated 3 years ago
• 4rtemi5 / imax
  Image augmentation library for Jax
  ☆37Updated 7 months ago
• f-dangel / vivit
  [TMLR 2022] Curvature access through the generalized Gauss-Newton's low-rank structure: Eigenvalues, eigenvectors, directional derivative…
  ☆17Updated last year
• aliutkus / torchsearchsorted
  Pytorch Custom CUDA kernel for searchsorted
  ☆135Updated last year
• yklcs / jaxsplat
  3D Gaussian Splatting in JAX
  ☆55Updated 5 months ago
• kazukiosawa / asdl
  ASDL: Automatic Second-order Differentiation Library for PyTorch
  ☆179Updated 3 months ago
• tmbdev-archive / webdataset-lightning
  A small demonstration of using WebDataset with ImageNet and PyTorch Lightning
  ☆73Updated 11 months ago
• hukkelas / sharpened_cosine_similarity_torch
  A Sharpened Cosine Similarity layer for PyTorch
  ☆10Updated 2 years ago
• paganpasta / eqxvision
  A Python package of computer vision models for the Equinox ecosystem.
  ☆102Updated 4 months ago
• ASEM000 / pytreeclass
  Visualize, create, and operate on pytrees in the most intuitive way possible.
  ☆41Updated last month
• tatsy / normalizing-flows-pytorch
  PyTorch implementations of normalizing flow and its variants.
  ☆74Updated 3 years ago