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:
- Implementation of several Generative Adversarial Networks in JAX / Flax☆34Updated 3 years ago
- Source code for Fathony, Sahu, Willmott, & Kolter, "Multiplicative Filter Networks", ICLR 2021.☆95Updated 4 years ago
- TorchFSM: Fourier Spectral Method with PyTorch☆44Updated 2 weeks ago
- The 2D discrete wavelet transform for JAX☆43Updated 2 years ago
- Principal Component Anlaysis (PCA) in PyTorch.☆22Updated last month
- A sparse KLU solver for PyTorch.☆65Updated 3 years ago
- FID computation in Jax/Flax.☆27Updated 11 months ago
- code associated with paper "Sparse Bayesian Optimization"☆26Updated last year
- JAX-SPH: A Differentiable Smoothed Particle Hydrodynamics Framework☆69Updated 5 months ago
- JAX-DIPS is a differentiable interfacial PDE solver.☆44Updated 9 months ago
- Artistic style transfer has been part of the quickly growing AI Art community in recent times. Pioneered by Gatys et al this class of met…☆27Updated 3 years ago
- PyTorch implementation of the Hessian-free optimizer☆35Updated last year
- Minimal JAX/Flax port of `lpips` supporting `vgg16`, with pre-trained weights stored in the 🤗 Hugging Face hub.☆15Updated 2 years ago
- Fourier Features Let Networks Learn High Frequency Functions in Low Dimensional Domains☆48Updated 5 years ago
- Geometry processing utilities compatible with jax for autodifferentiation.☆88Updated last year
- Unofficial implementation of 'Implicit Neural Representations with Periodic Activation Functions'☆52Updated 5 years ago
- Pytorch implementation of Generative Models as Distributions of Functions 🌿☆149Updated last year
- The ✨Magical✨ JAX ML Library.☆17Updated 5 months ago
- Code for the ICLR 2020 paper "Learning to Control PDEs"☆33Updated 5 years ago
- Intuitive scientific computing with dimension types for Jax, PyTorch, TensorFlow & NumPy☆90Updated last month
- a multiresolution convolutional autoencoder architecture☆22Updated 4 years ago
- Clifford-Steerable Convolutional Neural Networks [ICML'24]☆47Updated last month
- Official repo for AutoInt: Automatic Integration for Fast Neural Volume Rendering in CVPR 2021☆185Updated 2 years ago
- Code repository for "Learned Turbulence Modelling with Differentiable Fluid Solvers"☆38Updated 2 years ago
- Zonal Spherical Harmonics in d Dimensions in TensorFlow, PyTorch and Jax☆30Updated last year
- 🚀 A powerful library for efficient training of Neural Fields at scale.☆29Updated last year
- Official implementation of the AIAA Journal paper "Uncertainty-aware Surrogate Models for Airfoil Flow Simulations with Denoising Diffusi…☆68Updated 7 months ago
- A set of latex templates and TikZ/pgfplots figures☆17Updated last year
- ☆37Updated 4 years ago
- Pretrained deep learning models for Jax/Flax: StyleGAN2, GPT2, VGG, ResNet, etc.☆254Updated 3 months ago