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
• minimaxir / gpt-j-6b-experiments
  Test prompts for GPT-J-6B and the resulting AI-generated texts
  ☆53Updated 4 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
• nshepperd / lazy
  Lazy, a tool for running things in idle time
  ☆48Updated 4 years ago
• lucidrains / mlp-gpt-jax
  A GPT, made only of MLPs, in Jax
  ☆58Updated 4 years ago
• cair / PyTsetlinMachineCUDA
  Massively Parallel and Asynchronous Architecture for Logic-based AI
  ☆42Updated 2 years ago
• catid / dataloader
  High-performance tokenized language data-loader for Python C++ extension
  ☆13Updated last year
• EleutherAI / pyfra
  Python Research Framework
  ☆106Updated 2 years ago
• nimbo-sh / nimbo
  Run compute jobs on AWS as if you were running them locally.
  ☆125Updated 3 years ago
• outergroup / vexpr
  Write your code as tree-like expressions, then transform it
  ☆21Updated last year
• tmbdev-archive / tarproc
  Utilities for sequential processing of tar files.
  ☆24Updated 3 years ago
• crowsonkb / dice-mc
  DiCE: The Infinitely Differentiable Monte-Carlo Estimator
  ☆31Updated 2 years ago
• crowsonkb / esgd
  ESGD-M is a stochastic non-convex second order optimizer, suitable for training deep learning models, for PyTorch.
  ☆56Updated 2 years ago
• gdrtodd / ludax
  Ludax is a domain-specific language for board games that automatically compiles into hardware-accelerated learning environments with the …
  ☆22Updated 2 months ago
• RuiShu / tensorsketch
  A lightweight library for tensorflow 2.0
  ☆66Updated 5 years ago
• sergio-hcsoft / FractalAI
  Cellular automaton-based calculus for the masses
  ☆24Updated 7 years ago
• lucidrains / holodeck-pytorch
  Implementation of a holodeck, written in Pytorch
  ☆18Updated last year
• cgarciae / einop
  ☆60Updated 3 years ago
• srihari-humbarwadi / tpu_index
  TPU index is a package for fast similarity search over large collections of high dimension vectors on TPUs
  ☆17Updated 3 years ago
• RobinKa / jaxga
  Geometric Algebra package for JAX
  ☆50Updated 3 years ago
• mhlr / awesome-jax
  List of awesome JAX resources
  ☆13Updated 2 years ago
• namedtensor / notation
  ☆108Updated 2 years ago
• IpsumDominum / Pytorch-Simple-Transformer
  A simple transformer implementation without difficult syntax and extra bells and whistles.
  ☆53Updated 2 years ago
• NTT123 / pax
  A stateful pytree library for training neural networks.
  ☆22Updated 3 years ago
• breandan / picograd
  🦠 AD in less than 20 lines
  ☆54Updated 4 years ago
• ofnote / tsanley
  A runtime shape checker and auto-annotator for tensor programs (pronounced "stanley")
  ☆40Updated 5 years ago
• Popgun-Labs / PopGen
  A generative modelling toolkit for PyTorch.
  ☆70Updated 3 years ago
• AranKomat / Metroplex
  ☆21Updated 2 years ago
• nshepperd / jaxtorch
  A JAX nn library
  ☆21Updated 2 weeks ago
• jax-ml / australis
  ☆28Updated 2 years ago