jmtomczak / vae_kan_example
A simple example of VAEs with KANs
☆12Updated 5 months ago
Related projects ⓘ
Alternatives and complementary repositories for vae_kan_example
- Your favourite classical machine learning algos on the GPU/TPU☆20Updated last month
- Source code for the paper "Positional Attention: Out-of-Distribution Generalization and Expressivity for Neural Algorithmic Reasoning"☆14Updated last month
- Meta Flow Matching: Integrating Vector Fields on the Wasserstein Manifold☆39Updated last month
- Code repository for Trajectory Flow Matching☆21Updated last week
- Diffusion models in PyTorch☆86Updated 3 weeks ago
- Collection of tests performed during the study of the new Kolmogorov-Arnold Neural Networks (KAN)☆34Updated last month
- ☆14Updated last year
- Bare-bones implementations of some generative models in Jax: diffusion, normalizing flows, consistency models, flow matching, (beta)-VAEs…☆122Updated 10 months ago
- Automatic Integration for Neural Spatio-Temporal Point Process models (AI-STPP) is a new paradigm for exact, efficient, non-parametric inf…☆24Updated 3 weeks ago
- ☆27Updated 5 months ago
- "How to Trust Your Diffusion Models: A Convex Optimization Approach to Conformal Risk Control"☆16Updated last year
- ☆57Updated this week
- Graph neural networks in JAX.☆67Updated 4 months ago
- Kolmogorov-Arnold Networks (KAN) using orthogonal polynomials instead of B-splines.☆29Updated 5 months ago
- Generative Pretrained Autoregressive Transformer Graph Neural Network applied to the Analysis and Discovery of Materials☆17Updated last year
- The code for "Diffusion Geometry" (2024).☆26Updated 5 months ago
- Official source code for "Graph Neural Networks for Learning Equivariant Representations of Neural Networks". In ICLR 2024 (oral).☆70Updated 3 months ago
- Implementation of papers in 101 lines of code.☆18Updated 11 months ago
- Code for Paper "Think While You Generate: Discrete Diffusion with Planned Denoising"☆20Updated 3 weeks ago
- Pytorch implementation of a simple way to enable (Stochastic) Frame Averaging for any network☆46Updated 3 months ago
- This code implements a Radial Basis Function (RBF) based Kolmogorov-Arnold Network (KAN) for function approximation.☆25Updated 4 months ago
- Neural Optimal Transport with Lagrangian Costs☆47Updated 3 months ago
- Kolmogorov–Arnold Networks with modified activation (using MLP to represent the activation)☆104Updated last week
- High order and sparse layers in pytorch. Lagrange Polynomial, Piecewise Lagrange Polynomial, Piecewise Discontinuous Lagrange Polynomial…☆41Updated 4 months ago
- A minimal implementation of Equivariant Neural Fields (https://arxiv.org/abs/2406.05753).☆14Updated 2 months ago
- Official Implementation of the ICML 2023 paper: "Neural Wave Machines: Learning Spatiotemporally Structured Representations with Locally …☆69Updated last year
- Differentiable Euler Characteristic Transform☆17Updated 4 months ago
- Discrete Bayesian optimization with LLMs, PEFT finetuning methods, and the Laplace approximation.☆15Updated 3 months ago
- Free-form flows are a generative model training a pair of neural networks via maximum likelihood☆35Updated 4 months ago
- TopoBenchmark is a Python library designed to standardize benchmarking and accelerate research in Topological Deep Learning☆85Updated this week