Alternatives and similar repositories for jaxnotes

Users that are interested in jaxnotes are comparing it to the libraries listed below

• shawwn / ml-notes
  ☆39Updated 2 years ago
• nshepperd / lazy
  Lazy, a tool for running things in idle time
  ☆48Updated 4 years ago
• minimaxir / gpt-j-6b-experiments
  Test prompts for GPT-J-6B and the resulting AI-generated texts
  ☆53Updated 4 years ago
• EleutherAI / pyfra
  Python Research Framework
  ☆106Updated 2 years ago
• Rishit-dagli / Transformer-in-Transformer
  An Implementation of Transformer in Transformer in TensorFlow for image classification, attention inside local patches
  ☆43Updated 3 years ago
• Rishit-dagli / Nystromformer
  An implementation of the Nyströmformer, using Nystrom method to approximate standard self attention
  ☆56Updated 2 years ago
• alvarobartt / safejax
  Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`
  ☆45Updated last year
• kir-gadjello / zipslicer
  A library for incremental loading of large PyTorch checkpoints
  ☆56Updated 2 years ago
• cgarciae / einop
  ☆60Updated 3 years ago
• namedtensor / notation
  ☆108Updated 2 years ago
• naraB / colab-notifier
  A Google Chrome extension that notifies you when longer running Google Colaboraty cells are finished
  ☆24Updated 5 years ago
• crowsonkb / dice-mc
  DiCE: The Infinitely Differentiable Monte-Carlo Estimator
  ☆31Updated last year
• RobinKa / jaxga
  Geometric Algebra package for JAX
  ☆49Updated 3 years ago
• ofnote / tsanley
  A runtime shape checker and auto-annotator for tensor programs (pronounced "stanley")
  ☆40Updated 5 years ago
• lucidrains / mlp-gpt-jax
  A GPT, made only of MLPs, in Jax
  ☆58Updated 4 years ago
• sergio-hcsoft / FractalAI
  Cellular automaton-based calculus for the masses
  ☆24Updated 6 years ago
• LaihoE / did-it-spill
  Check if you have training samples in your test set
  ☆64Updated 3 years ago
• RuiShu / tensorsketch
  A lightweight library for tensorflow 2.0
  ☆66Updated 5 years ago
• jax-ml / australis
  ☆28Updated 2 years ago
• epfl-dlab / GPTurk
  ☆29Updated last year
• shawwn / scrap
  Nearly a thousand bash and python scripts I've written over the years.
  ☆123Updated 5 months ago
• noahtren / Freewire
  An experiment with "freely" wired neural networks (no layers)
  ☆163Updated 11 months ago
• n2cholas / shapecheck
  Framework-agnostic library for checking array/tensor shapes at runtime.
  ☆46Updated 4 years ago
• GallagherCommaJack / modulax
  ☆17Updated 10 months ago
• cair / PyTsetlinMachineCUDA
  Massively Parallel and Asynchronous Architecture for Logic-based AI
  ☆42Updated 2 years ago
• 8bitmp3 / JAX-Flax-Tutorial-Image-Classification-with-Linen
  How to use the Flax Linen API to build a convolutional neural network model and train it for image classification (using TensorFlow Datas…
  ☆24Updated last year
• libreai / neural-painters-x
  Neural Paiters
  ☆79Updated 4 years ago
• mhlr / awesome-jax
  List of awesome JAX resources
  ☆13Updated 2 years ago
• outergroup / vexpr
  Write your code as tree-like expressions, then transform it
  ☆21Updated last year
• mila-iqia / paperoni
  Search for scientific papers on the command line
  ☆104Updated this week