lweitkamp / GANs-JAX
Implementation of several Generative Adversarial Networks in JAX / Flax
☆33Updated 2 years ago
Related projects ⓘ
Alternatives and complementary repositories for GANs-JAX
- Pretrained deep learning models for Jax/Flax: StyleGAN2, GPT2, VGG, ResNet, etc.☆238Updated last year
- Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).☆104Updated 2 years ago
- Implementing the Denoising Diffusion Probabilistic Model in Flax☆142Updated 2 years ago
- Tensorflow 2.0 implementation of Fourier Feature Mapping Networks.☆42Updated 4 years ago
- Unofficial implementation of 'Implicit Neural Representations with Periodic Activation Functions'☆51Updated 4 years ago
- The 2D discrete wavelet transform for JAX☆38Updated last year
- Run PyTorch in JAX. 🤝☆201Updated this week
- FID computation in Jax/Flax.☆24Updated 4 months ago
- Texture mapping with variational auto-encoders☆40Updated 3 years ago
- JMP is a Mixed Precision library for JAX.☆187Updated 6 months ago
- Running Jax in PyTorch Lightning☆82Updated 2 weeks ago
- Toy implementations of some popular ML optimizers using Python/JAX☆43Updated 3 years ago
- Repo to the paper "Lie Point Symmetry Data Augmentation for Neural PDE Solvers"☆48Updated last year
- JAX-SPH: A Differentiable Smoothed Particle Hydrodynamics Framework☆54Updated 2 months ago
- Implementation of different GAN in JAX/Haiku.☆13Updated 2 years ago
- Contrastive Language-Image Pretraining☆143Updated 2 years ago
- Flax (JAX) implementation of Progressive Growing of GANs for Improved Quality, Stability, and Variation☆12Updated 3 years ago
- Source code for the ICLR'22 paper on "Half-Inverse Gradients"☆17Updated 2 years ago
- Reference implementation of Finite Element Networks as proposed in "Learning the Dynamics of Physical Systems from Sparse Observations wi…☆69Updated 6 months ago
- CLU lets you write beautiful training loops in JAX.☆321Updated last week
- Solver-in-the-Loop: Learning from Differentiable Physics to Interact with Iterative PDE-Solvers☆146Updated 2 years ago
- Implementation of Hourglass Transformer, in Pytorch, from Google and OpenAI☆84Updated 2 years ago
- ☆47Updated last year
- Implementation of the Adan (ADAptive Nesterov momentum algorithm) Optimizer in Pytorch☆247Updated 2 years ago
- PIX is an image processing library in JAX, for JAX.☆390Updated 2 weeks ago
- Image augmentation library for Jax☆37Updated 7 months ago
- Unofficial JAX implementations of deep learning research papers☆151Updated 2 years ago
- A criticism of a recent paper on buggy image downsampling methods in popular image processing and deep learning libraries.☆72Updated 2 years ago
- A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations☆122Updated 2 months ago