valentingol / GANJax

Related projects ⓘ

Alternatives and complementary repositories for GANJax

• lweitkamp / GANs-JAX
  Implementation of several Generative Adversarial Networks in JAX / Flax
  ☆33Updated 2 years ago
• scart97 / Siren-fastai2
  Unofficial implementation of 'Implicit Neural Representations with Periodic Activation Functions'
  ☆51Updated 4 years ago
• matthias-wright / jax-fid
  FID computation in Jax/Flax.
  ☆24Updated 4 months ago
• astanziola / fourier-neural-operator-flax
  An unofficial implementation of the Fourier Neural Operator in Flax
  ☆16Updated 6 months ago
• crowsonkb / jax-wavelets
  The 2D discrete wavelet transform for JAX
  ☆38Updated last year
• crowsonkb / mdmm
  The Modified Differential Multiplier Method (MDMM) for PyTorch
  ☆54Updated 3 years ago
• tumaer / jax-sph
  JAX-SPH: A Differentiable Smoothed Particle Hydrodynamics Framework
  ☆54Updated 2 months ago
• titu1994 / tf_fourier_features
  Tensorflow 2.0 implementation of Fourier Feature Mapping Networks.
  ☆42Updated 4 years ago
• patrick-kidger / optimistix
  Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/
  ☆334Updated last week
• yklcs / jaxsplat
  3D Gaussian Splatting in JAX
  ☆55Updated 5 months ago
• rubenohana / Gibbs-Diffusion
  Code of the paper "Listening to the Noise: Blind Denoising with Gibbs Diffusion"
  ☆29Updated 5 months ago
• ludwigwinkler / JaxLightning
  Running Jax in PyTorch Lightning
  ☆82Updated 2 weeks ago
• rasmusbergpalm / nanograd
  A miniscule implementation of reverse mode auto-differentiation
  ☆28Updated 3 years ago
• paganpasta / eqxvision
  A Python package of computer vision models for the Equinox ecosystem.
  ☆102Updated 4 months ago
• flaport / torch_sparse_solve
  A sparse KLU solver for PyTorch.
  ☆62Updated 2 years ago
• matthias-wright / flaxmodels
  Pretrained deep learning models for Jax/Flax: StyleGAN2, GPT2, VGG, ResNet, etc.
  ☆238Updated last year
• loliverhennigh / PhantomGaze
  Simple GPU rendering of scientific data with Pytorch, Jax, CuPy, and Warp backends.
  ☆21Updated last year
• halcy / tpuddim
  ☆22Updated 2 years ago
• shawwn / stylegan2
  StyleGAN2 - Official TensorFlow Implementation
  ☆25Updated 4 years ago
• clayton-r / symtorch
  Initial Commit
  ☆18Updated 3 years ago
• JAX-DIPS / JAX-DIPS
  JAX-DIPS is a differentiable interfacial PDE solver.
  ☆40Updated 2 months ago
• patil-suraj / stable-diffusion-jax
  ☆86Updated 2 years ago
• neuraloperator / GANO
  ☆31Updated last year
• mle-infrastructure / mle-hyperopt
  Lightweight Hyperparameter Optimization 🚂
  ☆143Updated 2 months ago
• rasmusbergpalm / vnca
  ☆47Updated last year
• kingoflolz / CLIP_JAX
  Contrastive Language-Image Pretraining
  ☆143Updated 2 years ago
• fastai / fastdiffusion
  Notes and plans for fastdiffusion course
  ☆202Updated last year
• boschresearch / multiplicative-filter-networks
  Source code for Fathony, Sahu, Willmott, & Kolter, "Multiplicative Filter Networks", ICLR 2021.
  ☆90Updated 3 years ago