Alternatives and similar repositories for vivit:

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

• f-dangel / singd
  [ICML 2024] SINGD: KFAC-like Structured Inverse-Free Natural Gradient Descent (http://arxiv.org/abs/2312.05705)
  ☆21Updated 4 months ago
• f-dangel / phd-thesis
  Source code for my PhD thesis: Backpropagation Beyond the Gradient
  ☆20Updated 2 years ago
• f-dangel / einconv
  Convolutions and more as einsum for PyTorch
  ☆14Updated 9 months ago
• SaitejaUtpala / rieoptax
  Riemannian Optimization Using JAX
  ☆48Updated last year
• ASEM000 / pytreeclass
  Visualize, create, and operate on pytrees in the most intuitive way possible.
  ☆44Updated 2 months ago
• f-dangel / sirfshampoo
  [ICML 2024] SIRFShampoo: Structured inverse- and root-free Shampoo in PyTorch (https://arxiv.org/abs/2402.03496)
  ☆14Updated 4 months ago
• glouppe / flowing-with-jax
  ☆15Updated 4 years ago
• AndPotap / einsum-search
  ☆30Updated 5 months ago
• andres-fr / skerch
  Sketched matrix decompositions for PyTorch
  ☆70Updated last month
• fKunstner / noise-sgd-adam-sign
  ☆16Updated last year
• cgarciae / einop
  ☆59Updated 3 years ago
• facebookresearch / rcpm
  Riemannian Convex Potential Maps
  ☆67Updated last year
• tychovdo / lila
  Code for "Invariance Learning in Deep Neural Networks with Differentiable Laplace Approximations"
  ☆23Updated 2 years ago
• f-dangel / curvlinops
  PyTorch linear operators for curvature matrices (Hessian, Fisher/GGN, KFAC, ...)
  ☆34Updated 2 weeks ago
• pnkraemer / matfree
  Matrix-free linear algebra in JAX.
  ☆116Updated 2 months ago
• necludov / action-matching
  Implementation of Action Matching for the Schrödinger equation
  ☆24Updated last year
• Artur-Galstyan / statedict2pytree
  ☆40Updated 3 months ago
• diffprog / code
  Code for the book "The Elements of Differentiable Programming".
  ☆76Updated this week
• facebookresearch / semi-discrete-flow
  code for "Semi-Discrete Normalizing Flows through Differentiable Tessellation"
  ☆26Updated 2 years ago
• facebookresearch / neural-conservation-law
  code for "Neural Conservation Laws A Divergence-Free Perspective".
  ☆37Updated 2 years ago
• patrick-kidger / quax
  Multiple dispatch over abstract array types in JAX.
  ☆115Updated 3 weeks ago
• COR-OPT / SuperPolyak.py
  Pytorch implementation of SuperPolyak subgradient method.
  ☆43Updated 2 years ago
• Information-Fusion-Lab-Umass / NuX
  Normalizing Flows using JAX
  ☆82Updated last year
• toshas / torch-householder
  Efficient Householder Transformation in PyTorch
  ☆64Updated 3 years ago
• katalinic / sdflows
  JAX exponential map normalising flows on sphere
  ☆17Updated 4 years ago
• imbirik / LieStationaryKernels
  Supplementary code for the paper "Stationary Kernels and Gaussian Processes on Lie Groups and their Homogeneous Spaces"
  ☆42Updated last year
• rdyro / torch2jax
  Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.
  ☆47Updated 3 weeks ago
• gisilvs / AEF
  ☆33Updated 2 years ago
• NeilGirdhar / tjax
  Tools for JAX
  ☆45Updated last week
• AaltoML / PeriodicBNN
  Code for 'Periodic Activation Functions Induce Stationarity' (NeurIPS 2021)
  ☆18Updated 3 years ago