Alternatives and similar repositories for vae_kan_example

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

• LiibanMo / scikit-jax
  Your favourite classical machine learning algos on the GPU/TPU
  ☆20Updated 2 weeks ago
• watcl-lab / positional_attention
  Source code for the paper "Positional Attention: Expressivity and Learnability of Algorithmic Computation"
  ☆14Updated 7 months ago
• smsharma / minified-generative-models
  Bare-bones implementations of some generative models in Jax: diffusion, normalizing flows, consistency models, flow matching, (beta)-VAEs…
  ☆141Updated 2 years ago
• probabilists / azula
  Diffusion models in PyTorch
  ☆120Updated last week
• maxxxzdn / erwin
  Erwin: A Tree-based Hierarchical Transformer for Large-scale Physical Systems [ICML'25]
  ☆109Updated 2 months ago
• itsdaniele / jeometric
  Graph neural networks in JAX.
  ☆68Updated last year
• lucidrains / frame-averaging-pytorch
  Pytorch implementation of a simple way to enable (Stochastic) Frame Averaging for any network
  ☆51Updated last year
• facebookresearch / adjoint_sampling
  code for "Adjoint Sampling: Highly Scalable Diffusion Samplers via Adjoint Matching"
  ☆130Updated 5 months ago
• maxxxzdn / clifford-group-equivariant-cnns
  Clifford-Steerable Convolutional Neural Networks [ICML'24]
  ☆51Updated 7 months ago
• uzulim / hades
  Fast singularity detection with kernel
  ☆37Updated last year
• Zhangyanbo / MLP-KAN
  Kolmogorov–Arnold Networks with modified activation (using MLP to represent the activation)
  ☆107Updated 2 months ago
• geometric-intelligence / TopoBench
  TopoBench is a Python library designed to standardize benchmarking and accelerate research in Topological Deep Learning
  ☆216Updated this week
• Iolo-Jones / ManifoldDiffusionGeometry
  Code for "Manifold Diffusion Geometry: Curvature, Tangent Spaces, and Dimension"
  ☆44Updated last year
• Rose-STL-Lab / AutoSTPP
  Automatic Integration for Neural Spatio-Temporal Point Process models (AI-STPP) is a new paradigm for exact, efficient, non-parametric inf…
  ☆25Updated last year
• njwfish / DistributionEmbeddings
  ☆34Updated 2 months ago
• nate-gillman / fourier-head
  Official implementation of "Fourier Head: Helping Large Language Models Learn Complex Probability Distributions" (ICLR 2025)
  ☆67Updated 9 months ago
• ml-jku / hopfield-boosting
  ☆33Updated last year
• nZhangx / TrajectoryFlowMatching
  Code repository for Trajectory Flow Matching
  ☆96Updated last year
• akandykeller / NeuralWaveMachines
  Official Implementation of the ICML 2023 paper: "Neural Wave Machines: Learning Spatiotemporally Structured Representations with Locally …
  ☆77Updated 2 years ago
• lazaratan / meta-flow-matching
  Meta Flow Matching: Integrating Vector Fields on the Wasserstein Manifold
  ☆69Updated 9 months ago
• tae898 / vae-diffusion
  ☆32Updated 5 months ago
• Iolo-Jones / DiffusionGeometry
  The code for "Diffusion Geometry" (2024).
  ☆44Updated last year
• facebookresearch / lagrangian-ot
  Neural Optimal Transport with Lagrangian Costs
  ☆60Updated 7 months ago
• jloveric / high-order-layers-torch
  High order and sparse layers in pytorch. Lagrange Polynomial, Piecewise Lagrange Polynomial, Piecewise Discontinuous Lagrange Polynomial…
  ☆44Updated last year
• AhmedImtiazPrio / grok-adversarial
  Deep Networks Grok All the Time and Here is Why
  ☆38Updated last year
• sidhu2690 / RBF-KAN
  This code implements a Radial Basis Function (RBF) based Kolmogorov-Arnold Network (KAN) for function approximation.
  ☆29Updated last year
• Rose-STL-Lab / Multi-Fidelity-Deep-Active-Learning
  ☆15Updated 2 years ago
• mcbal / spin-model-transformers
  Physics-inspired transformer modules based on mean-field dynamics of vector-spin models in JAX
  ☆45Updated 2 years ago
• pchlenski / manify
  A Python Library for Learning Non-Euclidean Representations
  ☆68Updated 4 months ago
• requeima / llm_processes
  ☆68Updated 9 months ago