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 7 months ago
• probabilists / azula
  Diffusion models in PyTorch
  ☆107Updated 2 months ago
• smsharma / minified-generative-models
  Bare-bones implementations of some generative models in Jax: diffusion, normalizing flows, consistency models, flow matching, (beta)-VAEs…
  ☆133Updated last year
• opallab / positional_attention
  Source code for the paper "Positional Attention: Expressivity and Learnability of Algorithmic Computation"
  ☆14Updated 2 months ago
• itsdaniele / jeometric
  Graph neural networks in JAX.
  ☆67Updated last year
• facebookresearch / adjoint_sampling
  code for "Adjoint Sampling: Highly Scalable Diffusion Samplers via Adjoint Matching"
  ☆116Updated 3 weeks ago
• maxxxzdn / erwin
  Erwin: A Tree-based Hierarchical Transformer for Large-scale Physical Systems [ICML'25]
  ☆101Updated last month
• uzulim / hades
  Fast singularity detection with kernel
  ☆36Updated last year
• Rose-STL-Lab / Multi-Fidelity-Deep-Active-Learning
  ☆15Updated 2 years ago
• lucidrains / frame-averaging-pytorch
  Pytorch implementation of a simple way to enable (Stochastic) Frame Averaging for any network
  ☆50Updated last year
• facebookresearch / lagrangian-ot
  Neural Optimal Transport with Lagrangian Costs
  ☆57Updated 3 months ago
• mcbal / spin-model-transformers
  Physics-inspired transformer modules based on mean-field dynamics of vector-spin models in JAX
  ☆41Updated last year
• david-knigge / enf-min-jax
  A minimal implementation of Equivariant Neural Fields (https://arxiv.org/abs/2406.05753).
  ☆25Updated 6 months ago
• 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…
  ☆24Updated 10 months ago
• nZhangx / TrajectoryFlowMatching
  Code repository for Trajectory Flow Matching
  ☆75Updated 9 months ago
• Iolo-Jones / ManifoldDiffusionGeometry
  Code for "Manifold Diffusion Geometry: Curvature, Tangent Spaces, and Dimension"
  ☆44Updated 9 months ago
• maxxxzdn / clifford-group-equivariant-cnns
  Clifford-Steerable Convolutional Neural Networks [ICML'24]
  ☆49Updated 3 months ago
• Zhangyanbo / MLP-KAN
  Kolmogorov–Arnold Networks with modified activation (using MLP to represent the activation)
  ☆106Updated 9 months ago
• lazaratan / meta-flow-matching
  Meta Flow Matching: Integrating Vector Fields on the Wasserstein Manifold
  ☆58Updated 5 months ago
• AhmedImtiazPrio / grok-adversarial
  Deep Networks Grok All the Time and Here is Why
  ☆37Updated last year
• ruke1ire / RTF
  A State-Space Model with Rational Transfer Function Representation.
  ☆79Updated last year
• necludov / wl-mechanics
  ☆31Updated last year
• juliusberner / sde_sampler
  Improved sampling via learned diffusions (ICLR2024) and an optimal control perspective on diffusion-based generative modeling (TMLR2024)
  ☆65Updated 5 months ago
• Iolo-Jones / DiffusionGeometry
  The code for "Diffusion Geometry" (2024).
  ☆43Updated last year
• njwfish / DistributionEmbeddings
  ☆28Updated 3 weeks ago
• nate-gillman / fourier-head
  Official implementation of "Fourier Head: Helping Large Language Models Learn Complex Probability Distributions" (ICLR 2025)
  ☆65Updated 4 months ago
• AndPotap / einsum-search
  ☆32Updated 10 months ago
• Sulam-Group / k-rcps
  "How to Trust Your Diffusion Models: A Convex Optimization Approach to Conformal Risk Control"
  ☆18Updated 2 months ago
• jloveric / high-order-layers-torch
  High order and sparse layers in pytorch. Lagrange Polynomial, Piecewise Lagrange Polynomial, Piecewise Discontinuous Lagrange Polynomial…
  ☆45Updated last year
• GFNOrg / GFlowNet-EM
  Code for GFlowNet-EM, a novel algorithm for fitting latent variable models with compositional latents and an intractable true posterior.
  ☆41Updated last year