Alternatives and similar repositories for jaxnotes

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

• shawwn / ml-notes
  ☆40Updated 2 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
• 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
• crowsonkb / dice-mc
  DiCE: The Infinitely Differentiable Monte-Carlo Estimator
  ☆31Updated 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
• cgarciae / einop
  ☆60Updated 3 years ago
• crowsonkb / esgd
  ESGD-M is a stochastic non-convex second order optimizer, suitable for training deep learning models, for PyTorch.
  ☆56Updated 3 years ago
• sergio-hcsoft / FractalAI
  Cellular automaton-based calculus for the masses
  ☆24Updated 7 years ago
• jax-ml / australis
  ☆28Updated 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 2 years ago
• lucidrains / holodeck-pytorch
  Implementation of a holodeck, written in Pytorch
  ☆18Updated last year
• EleutherAI / magiCARP
  One stop shop for all things carp
  ☆59Updated 3 years ago
• mariel-pettee / choreo-graph
  Learning latent graph representations of the dancing body with GNNs
  ☆28Updated 2 years ago
• nshepperd / jaxtorch
  A JAX nn library
  ☆21Updated 3 weeks ago
• kingoflolz / swarm-jax
  Swarm training framework using Haiku + JAX + Ray for layer parallel transformer language models on unreliable, heterogeneous nodes
  ☆241Updated 2 years ago
• RobinKa / jaxga
  Geometric Algebra package for JAX
  ☆51Updated 3 years ago
• benjamin-recht / mnist_1_pt_2
  1.2% test error on MNIST using only least squares and numpy calls.
  ☆19Updated 2 years ago
• catid / dataloader
  High-performance tokenized language data-loader for Python C++ extension
  ☆13Updated last year
• TheoCoombes / crawlingathome
  A client library for LAION's effort to filter CommonCrawl with CLIP, building a large scale image-text dataset.
  ☆32Updated 2 years ago
• 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
• 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
• NTT123 / pax
  A stateful pytree library for training neural networks.
  ☆22Updated last month
• namedtensor / notation
  ☆108Updated 2 years ago
• jmward01 / lmplay
  A playground to make it easy to try crazy things
  ☆33Updated 3 months ago
• tmbdev-archive / tarproc
  Utilities for sequential processing of tar files.
  ☆24Updated 3 years ago
• jxbz / nero
  👑 Pytorch code for the Nero optimiser.
  ☆20Updated 2 years ago
• RuiShu / tensorsketch
  A lightweight library for tensorflow 2.0
  ☆66Updated 5 years ago