jmtomczak / vae_kan_example
A simple example of VAEs with KANs
☆12Updated 11 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
- Source code for the paper "Positional Attention: Out-of-Distribution Generalization and Expressivity for Neural Algorithmic Reasoning"☆14Updated 3 months ago
- This code implements a Radial Basis Function (RBF) based Kolmogorov-Arnold Network (KAN) for function approximation.☆28Updated 10 months ago
- Your favourite classical machine learning algos on the GPU/TPU☆20Updated 4 months ago
- Neural Optimal Transport with Lagrangian Costs☆55Updated 9 months ago
- ☆15Updated last year
- ☆34Updated 4 months ago
- Graph neural networks in JAX.☆67Updated 10 months ago
- Implementation of papers in 101 lines of code.☆18Updated last year
- Automatic Integration for Neural Spatio-Temporal Point Process models (AI-STPP) is a new paradigm for exact, efficient, non-parametric inf…☆24Updated 6 months ago
- The code for "Diffusion Geometry" (2024).☆42Updated 11 months ago
- Code repository for Trajectory Flow Matching☆63Updated 6 months ago
- Erwin: A Tree-based Hierarchical Transformer for Large-scale Physical Systems☆48Updated this week
- ☆28Updated 3 months ago
- Kolmogorov-Arnold Networks with various basis functions like B-Splines, Fourier, Chebyshev, Wavelets etc☆33Updated 11 months ago
- ☆11Updated 2 months ago
- Diffusion models in PyTorch☆101Updated last month
- Generative Pretrained Autoregressive Transformer Graph Neural Network applied to the Analysis and Discovery of Materials☆15Updated last year
- ☆32Updated 7 months ago
- "How to Trust Your Diffusion Models: A Convex Optimization Approach to Conformal Risk Control"☆17Updated 2 months ago
- Kolmogorov-Arnold Networks (KAN) using orthogonal polynomials instead of B-splines.☆36Updated 5 months ago
- Code repository of the paper "Clifford-Steerable Convolutional Neural Networks"☆47Updated 8 months ago
- nanoGPT using Equinox☆13Updated 2 years ago
- Code for "Manifold Diffusion Geometry: Curvature, Tangent Spaces, and Dimension"☆44Updated 5 months ago
- Meta Flow Matching: Integrating Vector Fields on the Wasserstein Manifold☆53Updated 2 months ago
- Implementation of Action Matching for the Schrödinger equation☆24Updated last year
- Collection of tests performed during the study of the new Kolmogorov-Arnold Neural Networks (KAN)☆40Updated 2 months ago
- Stochastic trace estimation using JAX☆14Updated last month
- High order and sparse layers in pytorch. Lagrange Polynomial, Piecewise Lagrange Polynomial, Piecewise Discontinuous Lagrange Polynomial…☆44Updated 10 months ago
- Differentiable Euler Characteristic Transform☆17Updated 10 months ago
- A minimal implementation of Equivariant Neural Fields (https://arxiv.org/abs/2406.05753).☆23Updated 3 months ago