jmtomczak / vae_kan_example
A simple example of VAEs with KANs
☆12Updated 9 months ago
Alternatives and similar repositories for vae_kan_example:
Users that are interested in vae_kan_example are comparing it to the libraries listed below
- Automatic Integration for Neural Spatio-Temporal Point Process models (AI-STPP) is a new paradigm for exact, efficient, non-parametric inf…☆24Updated 4 months ago
- Source code for the paper "Positional Attention: Out-of-Distribution Generalization and Expressivity for Neural Algorithmic Reasoning"☆14Updated 2 weeks ago
- Your favourite classical machine learning algos on the GPU/TPU☆20Updated last month
- Meta Flow Matching: Integrating Vector Fields on the Wasserstein Manifold☆45Updated 5 months ago
- Neural Optimal Transport with Lagrangian Costs☆52Updated 7 months ago
- Code repository of the paper "Clifford-Steerable Convolutional Neural Networks"☆46Updated 6 months ago
- This code implements a Radial Basis Function (RBF) based Kolmogorov-Arnold Network (KAN) for function approximation.☆26Updated 8 months ago
- Kolmogorov-Arnold Networks (KAN) using orthogonal polynomials instead of B-splines.☆33Updated 2 months ago
- Diffusion models in PyTorch☆92Updated this week
- C++ and Cuda ops for fused FourierKAN☆75Updated 9 months ago
- The code for "Diffusion Geometry" (2024).☆39Updated 8 months ago
- "How to Trust Your Diffusion Models: A Convex Optimization Approach to Conformal Risk Control"☆17Updated last year
- ☆30Updated 9 months ago
- ☆32Updated last month
- Code repository for Trajectory Flow Matching☆52Updated 3 months ago
- Quantification of Uncertainty with Adversarial Models☆28Updated last year
- Official Implementation of the ICML 2023 paper: "Neural Wave Machines: Learning Spatiotemporally Structured Representations with Locally …☆69Updated last year
- Collection of tests performed during the study of the new Kolmogorov-Arnold Neural Networks (KAN)☆36Updated 4 months ago
- Implementation of Action Matching for the Schrödinger equation☆24Updated last year
- Code for GFlowNet-EM, a novel algorithm for fitting latent variable models with compositional latents and an intractable true posterior.☆41Updated last year
- ☆14Updated last year
- Graph neural networks in JAX.☆67Updated 8 months ago
- Pytorch implementation of a simple way to enable (Stochastic) Frame Averaging for any network☆49Updated 6 months ago
- ☆31Updated 9 months ago
- Combine B-Splines (BS) and Radial Basis Functions (RBF) in Kolmogorov-Arnold Networks (KANs)☆25Updated last month
- ☆22Updated last month
- ☆29Updated 4 months ago
- Fine-grained, dynamic control of neural network topology in JAX.☆21Updated last year
- Bare-bones implementations of some generative models in Jax: diffusion, normalizing flows, consistency models, flow matching, (beta)-VAEs…☆127Updated last year
- Kolmogorov-Arnold Networks with various basis functions like B-Splines, Fourier, Chebyshev, Wavelets etc☆32Updated 9 months ago