jloveric / high-order-layers-torch
High order and sparse layers in pytorch. Lagrange Polynomial, Piecewise Lagrange Polynomial, Piecewise Discontinuous Lagrange Polynomial (Chebyshev nodes) and Fourier Series layers of arbitrary order. Piecewise implementations could be thought of as a 1d grid (for each neuron) where each grid element is Lagrange polynomial. Both full connecte…
☆44Updated 9 months ago
Alternatives and similar repositories for high-order-layers-torch:
Users that are interested in high-order-layers-torch are comparing it to the libraries listed below
- Code repository of the paper "Clifford-Steerable Convolutional Neural Networks"☆47Updated 7 months ago
- A State-Space Model with Rational Transfer Function Representation.☆78Updated 10 months ago
- This code implements a Radial Basis Function (RBF) based Kolmogorov-Arnold Network (KAN) for function approximation.☆28Updated 9 months ago
- Official repository of Implicit Neural Convolutional Kernels for Steerable CNNs, Zhdanov et al.☆28Updated last month
- Kolmogorov-Arnold Networks with various basis functions like B-Splines, Fourier, Chebyshev, Wavelets etc☆32Updated 10 months ago
- Diffusion models in PyTorch☆99Updated this week
- Neural Optimal Transport with Lagrangian Costs☆55Updated 8 months ago
- Kolmogorov-Arnold Networks (KAN) using orthogonal polynomials instead of B-splines.☆35Updated 4 months ago
- Sequence Modeling with Multiresolution Convolutional Memory (ICML 2023)☆122Updated last year
- Kolmogorov–Arnold Networks with modified activation (using MLP to represent the activation)☆103Updated 5 months ago
- Flow-matching algorithms in JAX☆86Updated 7 months ago
- A simple example of VAEs with KANs☆12Updated 10 months ago
- ☆87Updated 9 months ago
- Pytorch implementation of a simple way to enable (Stochastic) Frame Averaging for any network☆49Updated 8 months ago
- Free-form flows are a generative model training a pair of neural networks via maximum likelihood☆43Updated 2 months ago
- ☆38Updated 10 months ago
- Kolmogorov-Arnold networks (KAN) as implicit functions (like NeRF but simpler)☆14Updated 10 months ago
- ☆16Updated 3 years ago
- C++ and Cuda ops for fused FourierKAN☆76Updated 10 months ago
- ☆52Updated 5 months ago
- Repo to the paper "Lie Point Symmetry Data Augmentation for Neural PDE Solvers"☆49Updated last year
- The 2D discrete wavelet transform for JAX☆41Updated 2 years ago
- Graph neural networks in JAX.☆67Updated 9 months ago
- ☆172Updated 3 months ago
- Lightning-like training API for JAX with Flax☆38Updated 3 months ago
- Implementation of the proposed Adam-atan2 from Google Deepmind in Pytorch☆102Updated 4 months ago
- A practical implementation of GradNorm, Gradient Normalization for Adaptive Loss Balancing, in Pytorch☆88Updated last year
- Improved sampling via learned diffusions (ICLR2024) and an optimal control perspective on diffusion-based generative modeling (TMLR2024)☆61Updated 2 weeks ago
- Implementation of Gradient Agreement Filtering, from Chaubard et al. of Stanford, but for single machine microbatches, in Pytorch☆23Updated 2 months ago
- The Gaussian Histogram Loss (HL-Gauss) proposed by Imani et al. with a few convenient wrappers for regression, in Pytorch☆57Updated last month