Alternatives and similar repositories for jax-tutorial

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

• srush / jax-lda
  ☆30Updated 3 years ago
• shreyansh26 / ML-Optimizers-JAX
  Toy implementations of some popular ML optimizers using Python/JAX
  ☆44Updated 4 years ago
• prajjwal1 / fluence
  A deep learning library based on Pytorch focussed on low resource language research and robustness
  ☆70Updated 3 years ago
• cgraywang / transformer-on-diet
  Code repo for "Transformer on a Diet" paper
  ☆31Updated 5 years ago
• huggingface / model_card
  ☆30Updated 4 years ago
• fastai / fastpy
  An easy way to start a python programming environment using GitHub Codespaces.
  ☆15Updated 5 years ago
• mblondel / projection-losses
  Python implementation of projection losses.
  ☆27Updated 5 years ago
• sscardapane / hydra-tutorial
  Companion code for a tutorial on using Hydra.
  ☆30Updated 4 years ago
• widmi / widis-lstm-tools
  Tools for working with Long Short-Term Memory (LSTM) networks and sequences in Pytorch
  ☆36Updated 4 years ago
• fsschneider / cockpit-experiments
  Experiments for the NeurIPS 2021 paper "Cockpit: A Practical Debugging Tool for the Training of Deep Neural Networks"
  ☆13Updated 4 years ago
• ExpectationMax / sklearn-jax-kernels
  Composable kernels for scikit-learn implemented in JAX.
  ☆45Updated 5 years ago
• michalwols / yann
  Yet Another Neural Network Library 🤔
  ☆27Updated last month
• ethanluoyc / dhmc-jax
  Discontinuous Hamiltonian Monte Carlo in JAX
  ☆42Updated 5 years ago
• xflash96 / pybind11_package_example
  Example python package with pybind11 cpp extension
  ☆57Updated 4 years ago
• ColinQiyangLi / AdaCat
  AdaCat
  ☆49Updated 3 years ago
• MathInf / toroidal
  a lightweight transformer library for PyTorch
  ☆72Updated 3 years ago
• namedtensor / notation
  ☆108Updated 2 years ago
• karalets / TyXe
  Library for Bayesian Neural Networks in PyTorch (first version as published in ProbProg2020)
  ☆42Updated 4 years ago
• nvecoven / BRC
  A repository containing the code for the Bistable Recurrent Cell
  ☆47Updated 4 years ago
• nyu-mll / pretraining-learning-curves
  The repository for the paper "When Do You Need Billions of Words of Pretraining Data?"
  ☆21Updated 4 years ago
• cambridge-mlg / expressiveness-approx-bnns
  Code for "On the Expressiveness of Approximate Inference in Bayesian Neural Networks"
  ☆13Updated 4 years ago
• ahthie7u / cockpit
  Code for the anonymous submission "Cockpit: A Practical Debugging Tool for Training Deep Neural Networks"
  ☆31Updated 4 years ago
• Popgun-Labs / PopGen
  A generative modelling toolkit for PyTorch.
  ☆70Updated 4 years ago
• yngtdd / hyperspace
  Distributed Bayesian Optimization
  ☆23Updated 5 years ago
• cambridgeltl / parameter-factorization
  Factorization of the neural parameter space for zero-shot multi-lingual and multi-task transfer
  ☆39Updated 5 years ago
• antofuller / configaformers
  A python library for highly configurable transformers - easing model architecture search and experimentation.
  ☆49Updated 3 years ago
• srush / VirtualTeaching
  DIY setup for virtual teaching on ubuntu
  ☆38Updated 5 years ago
• bgroenks96 / normalizing-flows
  Implementations of normalizing flows using python and tensorflow
  ☆24Updated 10 months ago
• n2cholas / shapecheck
  Framework-agnostic library for checking array/tensor shapes at runtime.
  ☆46Updated 4 years ago
• pytorch-tpu / examples
  This repository contains example code to build models on TPUs
  ☆30Updated 2 years ago