Alternatives and similar repositories for jax-tutorial:

Users that are interested in jax-tutorial are comparing it to the libraries listed below

• hadyelsahar / EBM-visualizations
  notebooks of cool EBM visualizations
  ☆15Updated 4 years ago
• sscardapane / hydra-tutorial
  Companion code for a tutorial on using Hydra.
  ☆29Updated 3 years ago
• apmoore1 / fiesta
  Fast IdEntification of State-of-The-Art models using adaptive bandit algorithms
  ☆14Updated 2 years ago
• shreyansh26 / ML-Optimizers-JAX
  Toy implementations of some popular ML optimizers using Python/JAX
  ☆44Updated 3 years ago
• srush / jax-lda
  ☆30Updated 3 years ago
• michalwols / yann
  Yet Another Neural Network Library 🤔
  ☆27Updated last month
• twidddj / tokmangan
  codes for TokenManipulationGAN
  ☆7Updated 4 years ago
• ethanluoyc / dhmc-jax
  Discontinuous Hamiltonian Monte Carlo in JAX
  ☆41Updated 5 years ago
• harvardnlp / hmm-lm
  ☆42Updated 4 years ago
• karalets / TyXe
  Library for Bayesian Neural Networks in PyTorch (first version as published in ProbProg2020)
  ☆42Updated 3 years ago
• arogozhnikov / readable_capsnet
  Blazingly fast capsule networks in 75 lines of pytorch+einops
  ☆26Updated 3 years ago
• lucidrains / local-attention-flax
  Local Attention - Flax module for Jax
  ☆20Updated 3 years ago
• thiery-lab / manifold-mcmc-for-diffusions
  Code accompanying the paper 'Manifold MCMC methods for Bayesian inference in a wide class of diffusion models'
  ☆10Updated last year
• juliuskunze / cwvae-jax
  Clockwork VAEs in JAX/Flax
  ☆32Updated 3 years ago
• YeonwooSung / GLOM
  PyTorch implementation of GLOM
  ☆22Updated 3 years ago
• locuslab / ase
  Analogous Safe-state Exploration (ASE) is an algorithm for provably safe and optimal exploration in MDPs with unknown, stochastic dynamic…
  ☆11Updated 4 years ago
• aleatory-science / deep-probprog-course
  Deep Probabilistic Programming Course @ DIKU
  ☆57Updated 4 years ago
• MonaNashaat / Asterisk
  Generating Training Data Made Easy
  ☆43Updated 4 years ago
• yngtdd / hyperspace
  Distributed Bayesian Optimization
  ☆23Updated 4 years ago
• fastai / fastpy
  An easy way to start a python programming environment using GitHub Codespaces.
  ☆15Updated 4 years ago
• mblondel / projection-losses
  Python implementation of projection losses.
  ☆26Updated 5 years ago
• remilepriol / causal-adaptation-speed
  Investigate the speed of adaptation of structural causal models
  ☆16Updated 4 years ago
• franrruiz / uivi
  Code for Unbiased Implicit Variational Inference (UIVI)
  ☆14Updated 6 years ago
• samiraabnar / Reflect
  Official Implementation of "Transferring Inductive Biases Through Knowledge Distillation"
  ☆14Updated 4 years ago
• huggingface / model_card
  ☆30Updated 3 years ago
• gebob19 / introduction_to_normalizing_flows
  Jupyter Notebook corresponding to 'Going with the Flow: An Introduction to Normalizing Flows'
  ☆26Updated 4 years ago
• Popgun-Labs / PopGen
  A generative modelling toolkit for PyTorch.
  ☆70Updated 3 years ago
• iKernels / transformers-lightning
  A collection of Models, Datasets, DataModules, Callbacks, Metrics, Losses and Loggers to better integrate pytorch-lightning with transfor…
  ☆47Updated last year
• PAIR-code / jax-recommenders
  ☆9Updated 2 years ago
• ItzikMalkiel / MTAdam
  MTAdam: Automatic Balancing of Multiple Training Loss Terms
  ☆36Updated 4 years ago