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
• nshepperd / lazy
  Lazy, a tool for running things in idle time
  ☆48Updated 4 years ago
• EleutherAI / pyfra
  Python Research Framework
  ☆106Updated 2 years ago
• RobinKa / jaxga
  Geometric Algebra package for JAX
  ☆50Updated 3 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
• sergio-hcsoft / FractalAI
  Cellular automaton-based calculus for the masses
  ☆24Updated 7 years ago
• nimbo-sh / nimbo
  Run compute jobs on AWS as if you were running them locally.
  ☆125Updated 3 years ago
• ofnote / tsanley
  A runtime shape checker and auto-annotator for tensor programs (pronounced "stanley")
  ☆40Updated 5 years ago
• crowsonkb / dice-mc
  DiCE: The Infinitely Differentiable Monte-Carlo Estimator
  ☆31Updated 2 years ago
• Rishit-dagli / Nystromformer
  An implementation of the Nyströmformer, using Nystrom method to approximate standard self attention
  ☆56Updated 3 years ago
• tmbdev-archive / tarproc
  Utilities for sequential processing of tar files.
  ☆24Updated 3 years ago
• NTT123 / pax
  A stateful pytree library for training neural networks.
  ☆22Updated 3 years ago
• cgarciae / einop
  ☆60Updated 3 years ago
• namedtensor / notation
  ☆108Updated 2 years ago
• lucidrains / mlp-gpt-jax
  A GPT, made only of MLPs, in Jax
  ☆58Updated 4 years ago
• IpsumDominum / Pytorch-Simple-Transformer
  A simple transformer implementation without difficult syntax and extra bells and whistles.
  ☆53Updated 2 years ago
• sytelus / pcprep
  Various handy scripts to quickly setup new Linux and Windows sandboxes, containers and WSL.
  ☆40Updated 4 months ago
• EleutherAI / equivariance
  A framework for implementing equivariant DL
  ☆10Updated 4 years ago
• Popgun-Labs / PopGen
  A generative modelling toolkit for PyTorch.
  ☆70Updated 3 years ago
• jax-ml / australis
  ☆28Updated 2 years ago
• catid / dataloader
  High-performance tokenized language data-loader for Python C++ extension
  ☆13Updated last year
• glassroom / heinsen_routing
  Reference implementation of "An Algorithm for Routing Vectors in Sequences" (Heinsen, 2022) and "An Algorithm for Routing Capsules in All…
  ☆172Updated 2 years ago
• gdrtodd / ludax
  Ludax is a domain-specific language for board games that automatically compiles into hardware-accelerated learning environments with the …
  ☆23Updated last week
• EleutherAI / magiCARP
  One stop shop for all things carp
  ☆59Updated 2 years ago
• alvarobartt / safejax
  Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`
  ☆45Updated last year
• crowsonkb / esgd
  ESGD-M is a stochastic non-convex second order optimizer, suitable for training deep learning models, for PyTorch.
  ☆56Updated 2 years ago
• AranKomat / Metroplex
  ☆21Updated 2 years ago
• DAIOS-AI / mindscript
  A programming language for formal/informal computation.
  ☆41Updated 2 weeks ago
• HazyResearch / epoxy
  Interactive Model Iteration with Weak Supervision and Pre-Trained Embeddings
  ☆77Updated 3 years ago