valentingol / GANJax
Implementation of different GAN in JAX/Haiku.
☆14Updated 2 years ago
Alternatives and similar repositories for GANJax:
Users that are interested in GANJax are comparing it to the libraries listed below
- Implementation of several Generative Adversarial Networks in JAX / Flax☆33Updated 2 years ago
- The 2D discrete wavelet transform for JAX☆40Updated last year
- Reference implementation of Finite Element Networks as proposed in "Learning the Dynamics of Physical Systems from Sparse Observations wi…☆70Updated 8 months ago
- A tool for generating PDEs ground truth datasets from ARCSim, FEniCS and SU2☆35Updated 3 years ago
- A sparse KLU solver for PyTorch.☆62Updated 3 years ago
- JAX-SPH: A Differentiable Smoothed Particle Hydrodynamics Framework☆58Updated 2 weeks ago
- ICML 2023: Reduce, Reuse, Recycle: Composing Energy-Based Diffusion Models with MCMC☆135Updated 3 months ago
- Principal Component Anlaysis (PCA) in PyTorch.☆12Updated 2 months ago
- Code for "Journey to the BAOAB-limit: finding effective MCMC samplers for score-based models". See more at https://ajayj.com/journey.☆12Updated 2 years 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…☆24Updated 2 years ago
- Scale-invariant Learning by Physics Inversion (NeurIPS 2022)☆10Updated 2 years ago
- JAX-DIPS is a differentiable interfacial PDE solver.☆41Updated 4 months ago
- Official PyTorch implementation of the Vectorized Conditional Neural Field.☆12Updated 6 months ago
- An extensible benchmark suite to evaluate data-driven physical simulation☆25Updated 3 years ago
- Source code for the ICLR'22 paper on "Half-Inverse Gradients"☆18Updated 2 years ago
- [NeurIPS 2024] Official implementation of "NeuralClothSim: Neural Deformation Fields Meet the Thin Shell Theory"☆30Updated 3 months ago
- ☆14Updated last year
- code associated with paper "Sparse Bayesian Optimization"☆26Updated last year
- VectorAdam for Rotation Equivariant Geometry Optimization☆112Updated 2 years ago
- ☆21Updated 7 months ago
- 🚀 A powerful library for efficient training of Neural Fields at scale.☆28Updated last year
- Fusing Taichi into PyTorch☆141Updated 2 years ago
- ☆33Updated last year
- PyTorch implementation of the Hessian-free optimizer☆33Updated 7 months ago
- Official repo for AutoInt: Automatic Integration for Fast Neural Volume Rendering in CVPR 2021☆184Updated 2 years ago
- 3D Gaussian Splatting in JAX☆55Updated 8 months ago
- Source code for Fathony, Sahu, Willmott, & Kolter, "Multiplicative Filter Networks", ICLR 2021.☆92Updated 3 years ago
- Official Implementation of the paper: A Complete Recipe for Diffusion Generative Models☆30Updated 3 months ago
- Code for the book "The Elements of Differentiable Programming".☆72Updated 5 months ago
- v objective diffusion inference code for JAX.☆212Updated 2 years ago