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
• ExpectationMax / sklearn-jax-kernels
  Composable kernels for scikit-learn implemented in JAX.
  ☆46Updated 5 years ago
• prajjwal1 / fluence
  A deep learning library based on Pytorch focussed on low resource language research and robustness
  ☆70Updated 4 years ago
• lvwerra / deep-math
  Implementation of "Analysing Mathematical Reasoning Abilities of Neural Models"
  ☆30Updated 2 years ago
• cgraywang / transformer-on-diet
  Code repo for "Transformer on a Diet" paper
  ☆31Updated 5 years ago
• widmi / widis-lstm-tools
  Tools for working with Long Short-Term Memory (LSTM) networks and sequences in Pytorch
  ☆36Updated 4 years ago
• shreyansh26 / ML-Optimizers-JAX
  Toy implementations of some popular ML optimizers using Python/JAX
  ☆44Updated 4 years ago
• ethanluoyc / dhmc-jax
  Discontinuous Hamiltonian Monte Carlo in JAX
  ☆42Updated 5 years ago
• huggingface / model_card
  ☆30Updated 4 years ago
• hadyelsahar / EBM-visualizations
  notebooks of cool EBM visualizations
  ☆15Updated 4 years ago
• edwardjhu / improved_wasserstein
  Code for our ICLR Trustworthy ML 2020 workshop paper "Improved Image Wasserstein Attacks and Defenses"
  ☆14Updated 5 years ago
• google-research / precondition
  ☆31Updated this week
• MathInf / toroidal
  a lightweight transformer library for PyTorch
  ☆72Updated 4 years ago
• n2cholas / shapecheck
  Framework-agnostic library for checking array/tensor shapes at runtime.
  ☆46Updated 4 years ago
• srush / VirtualTeaching
  DIY setup for virtual teaching on ubuntu
  ☆38Updated 5 years ago
• probml / JSL
  Jax SSM Library
  ☆48Updated 3 years ago
• ganguli-lab / degrees-of-freedom
  ☆37Updated 3 years ago
• yngtdd / hyperspace
  Distributed Bayesian Optimization
  ☆23Updated 5 years ago
• remilepriol / causal-adaptation-speed
  Investigate the speed of adaptation of structural causal models
  ☆15Updated 4 years ago
• j-cunanan / MLT-contributions
  ☆22Updated 5 years ago
• namedtensor / notation
  ☆108Updated 3 years ago
• kyunghyuncho / backprop-kalman-filter
  ☆45Updated 6 years ago
• stanislavfort / adversaries_to_convnext
  Adversarial examples to the new ConvNeXt architecture
  ☆20Updated 4 years ago
• Popgun-Labs / PopGen
  A generative modelling toolkit for PyTorch.
  ☆70Updated 4 years ago
• SCCH-KVS / training-engine
  The stand-alone training engine module for the ALOHA.eu project.
  ☆15Updated 6 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
• hanneshapke / Deploying-Deep-Learning-Models
  Strategies to deploy deep learning models
  ☆27Updated 7 years ago
• harvardnlp / hmm-lm
  ☆40Updated 4 years ago
• inkawhich / pt-distributed-tutorial
  ☆17Updated 7 years ago
• google / jax-recommenders
  ☆11Updated 3 years ago