albanie / yaspi

Related projects ⓘ

Alternatives and complementary repositories for yaspi

• rolandgvc / flaxvision
  A selection of neural network models ported from torchvision for JAX & Flax.
  ☆44Updated 3 years ago
• choidami / sst
  ☆49Updated 4 years ago
• google-deepmind / einshape
  ☆99Updated 4 months ago
• NERSC / slurm-ray-cluster
  ☆52Updated last year
• j-towns / vdvae-jax
  Very deep VAEs in JAX/Flax
  ☆45Updated 3 years ago
• cgarciae / treeo
  A small library for creating and manipulating custom JAX Pytree classes
  ☆57Updated last year
• facebookresearch / semi-discrete-flow
  code for "Semi-Discrete Normalizing Flows through Differentiable Tessellation"
  ☆24Updated last year
• ml-research / pau
  Padé Activation Units: End-to-end Learning of Activation Functions in Deep Neural Network
  ☆64Updated 3 years ago
• AllanYangZhou / metalearning-symmetries
  Experiments for Meta-Learning Symmetries by Reparameterization
  ☆56Updated 3 years ago
• jxbz / fromage
  🧀 Pytorch code for the Fromage optimiser.
  ☆122Updated 4 months ago
• akbir / deq-jax
  [NeurIPS'19] Deep Equilibrium Models Jax Implementation
  ☆37Updated 4 years ago
• nicola-decao / power_spherical
  Pytorch implementation of the Power Spherical distribution
  ☆73Updated 4 months ago
• johnhw / differentiable_sorting
  Differentiable bitonic sorting
  ☆138Updated 4 years ago
• juliuskunze / cwvae-jax
  Clockwork VAEs in JAX/Flax
  ☆32Updated 3 years ago
• gbaydin / difftorch
  A differentiation API for PyTorch
  ☆30Updated 4 years ago
• jxbz / madam
  👩 Pytorch and Jax code for the Madam optimiser.
  ☆51Updated 3 years ago
• wouterkool / estimating-gradients-without-replacement
  Estimating Gradients for Discrete Random Variables by Sampling without Replacement
  ☆39Updated 4 years ago
• unixpickle / vae-textures
  Texture mapping with variational auto-encoders
  ☆40Updated 3 years ago
• gehring / fax
  ☆78Updated 3 years ago
• cunningham-lab / cb_and_cc
  ☆36Updated last year
• hardmaru / mdn_jax_tutorial
  Mixture Density Networks (Bishop, 1994) tutorial in JAX
  ☆58Updated 4 years ago
• pengzhenghao / use-ray-with-slurm
  A brief tutorial on how to use ray/rllib/tun in slurm cluster.
  ☆32Updated 3 years ago
• ericjang / nf-jax
  Normalizing Flows in Jax
  ☆105Updated 4 years ago
• plai-group / tvo
  Code for the Thermodynamic Variational Objective
  ☆26Updated 2 years ago
• srush / torch-queue
  ☆67Updated last year
• cornellius-gp / linear_operator.old
  A LinearOperator implementation for PyTorch
  ☆18Updated 3 years ago
• yobibyte / pgn
  Graph Nets in pytorch
  ☆27Updated last year
• pemami4911 / EfficientMORL
  EfficientMORL (ICML'21)
  ☆22Updated 3 years ago
• conormdurkan / autoregressive-energy-machines
  Autoregressive Energy Machines
  ☆77Updated 2 years ago
• srush / parallax
  ☆155Updated 4 years ago