abarcel / haikumodels

Related projects ⓘ

Alternatives and complementary repositories for haikumodels

• ColinQiyangLi / AdaCat
  AdaCat
  ☆49Updated 2 years ago
• khdlr / augmax
  Efficiently Composable Data Augmentation on the GPU with Jax
  ☆27Updated 3 months ago
• cgarciae / ciclo
  A functional training loops library for JAX
  ☆85Updated 8 months ago
• rolandgvc / flaxvision
  A selection of neural network models ported from torchvision for JAX & Flax.
  ☆44Updated 3 years ago
• 4rtemi5 / imax
  Image augmentation library for Jax
  ☆37Updated 7 months ago
• lixilinx / psgd_torch
  Pytorch implementation of preconditioned stochastic gradient descent (affine group preconditioner, low-rank approximation preconditioner …
  ☆110Updated 3 weeks ago
• cunningham-lab / cb_and_cc
  ☆36Updated last year
• cgarciae / treeo
  A small library for creating and manipulating custom JAX Pytree classes
  ☆58Updated last year
• gebob19 / introduction_to_normalizing_flows
  Jupyter Notebook corresponding to 'Going with the Flow: An Introduction to Normalizing Flows'
  ☆25Updated 3 years ago
• gisilvs / AEF
  ☆33Updated last year
• samuela / torch2jax
  Run PyTorch in JAX. 🤝
  ☆199Updated last year
• PrincetonLIPS / RandomizedAutomaticDifferentiation
  Experiment code for "Randomized Automatic Differentiation"
  ☆67Updated 4 years ago
• j-towns / vdvae-jax
  Very deep VAEs in JAX/Flax
  ☆45Updated 3 years ago
• n2cholas / jax-resnet
  Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).
  ☆104Updated 2 years ago
• mle-infrastructure / mle-scheduler
  Lightweight Cluster/Cloud VM Job Management 🚀
  ☆41Updated 2 months ago
• srush / torch-queue
  ☆66Updated last year
• optimizedlearning / mechanic
  ☆35Updated 9 months ago
• akbir / deq-jax
  [NeurIPS'19] Deep Equilibrium Models Jax Implementation
  ☆37Updated 4 years ago
• shyamsn97 / hyper-nn
  Easy Hypernetworks in Pytorch and Jax
  ☆96Updated last year
• brentyi / jax_dataclasses
  Pytrees + dataclasses ❤️
  ☆61Updated 4 months ago
• jrmcornish / cif
  PyTorch implementation of Continuously Indexed Flows paper, with many baseline normalising flows
  ☆30Updated 3 years ago
• cgarciae / einop
  ☆58Updated 2 years ago
• Information-Fusion-Lab-Umass / NuX
  Normalizing Flows using JAX
  ☆82Updated 11 months ago
• gehring / fax
  ☆78Updated 3 years ago
• paganpasta / eqxvision
  A Python package of computer vision models for the Equinox ecosystem.
  ☆102Updated 3 months ago
• facebookresearch / rcpm
  Riemannian Convex Potential Maps
  ☆68Updated last year
• unixpickle / vae-textures
  Texture mapping with variational auto-encoders
  ☆40Updated 3 years ago
• f-dangel / sirfshampoo
  [ICML 2024] SIRFShampoo: Structured inverse- and root-free Shampoo in PyTorch (https://arxiv.org/abs/2402.03496)
  ☆13Updated last week
• google-deepmind / jaxline
  ☆152Updated 10 months ago
• dylandoblar / noether-networks
  Meta-learning inductive biases in the form of useful conserved quantities.
  ☆37Updated last year