valentingol / GANJaxLinks
Implementation of different GAN in JAX/Haiku.
☆14Updated 3 years ago
Alternatives and similar repositories for GANJax
Users that are interested in GANJax are comparing it to the libraries listed below
Sorting:
- The 2D discrete wavelet transform for JAX☆44Updated 2 years ago
- Source code for the ICLR'22 paper on "Half-Inverse Gradients"☆19Updated 3 years ago
- 🚀 A powerful library for efficient training of Neural Fields at scale.☆29Updated last year
- Source code for Fathony, Sahu, Willmott, & Kolter, "Multiplicative Filter Networks", ICLR 2021.☆96Updated 4 years ago
- Implementation of several Generative Adversarial Networks in JAX / Flax☆35Updated 3 years ago
- code associated with paper "Sparse Bayesian Optimization"☆26Updated 2 years ago
- A parallel ODE solver for PyTorch☆274Updated last year
- Implicit^2: Implicit model for implicit neural representations☆29Updated 4 years ago
- Code for Point-Calibrated Spectral Neural Operators☆21Updated last year
- Develop C++/CUDA extensions with PyTorch like Python scripts☆10Updated 3 weeks ago
- ☆16Updated 2 years ago
- FID computation in Jax/Flax.☆29Updated last year
- Interpolation routines in Pytorch.☆111Updated 5 years ago
- ☆160Updated last year
- A sparse KLU solver for PyTorch.☆66Updated 4 years ago
- PyTorch implementation of the Hessian-free optimizer☆36Updated last year
- Fourier Features Let Networks Learn High Frequency Functions in Low Dimensional Domains☆51Updated 5 years ago
- The Modified Differential Multiplier Method (MDMM) for PyTorch☆60Updated 4 years ago
- Interpolating natural cubic splines. Includes batching, GPU support, support for missing values, evaluating derivatives of the spline, an…☆274Updated 3 years ago
- ☆16Updated 4 years ago
- Gradient-based constrained optimization for JAX☆37Updated 3 years ago
- High order and sparse layers in pytorch. Lagrange Polynomial, Piecewise Lagrange Polynomial, Piecewise Discontinuous Lagrange Polynomial…☆44Updated last year
- Texture mapping with variational auto-encoders☆40Updated 4 years ago
- Principal Component Anlaysis (PCA) in PyTorch.☆38Updated 6 months ago
- [TMLR 2022] Curvature access through the generalized Gauss-Newton's low-rank structure: Eigenvalues, eigenvectors, directional derivative…☆17Updated 2 years ago
- ☆33Updated 2 years ago
- An unofficial pytorch implementation of 《Fourier Features Let Networks Learn High Frequency Functions in Low Dimensional Domains》☆67Updated 5 years ago
- ☆17Updated last year
- TensorLy-Torch: Deep Tensor Learning with TensorLy and PyTorch☆82Updated last year
- Kolmogorov-Arnold networks (KAN) as implicit functions (like NeRF but simpler)☆14Updated last year