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
• 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
• cgarciae / einop
  ☆60Updated 3 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
• crowsonkb / dice-mc
  DiCE: The Infinitely Differentiable Monte-Carlo Estimator
  ☆31Updated 2 years ago
• Popgun-Labs / PopGen
  A generative modelling toolkit for PyTorch.
  ☆70Updated 3 years ago
• namedtensor / notation
  ☆108Updated 2 years ago
• outergroup / vexpr
  Write your code as tree-like expressions, then transform it
  ☆21Updated last year
• RuiShu / tensorsketch
  A lightweight library for tensorflow 2.0
  ☆66Updated 5 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
• RobinKa / jaxga
  Geometric Algebra package for JAX
  ☆50Updated 3 years ago
• EleutherAI / magiCARP
  One stop shop for all things carp
  ☆59Updated 3 years ago
• sergio-hcsoft / FractalAI
  Cellular automaton-based calculus for the masses
  ☆24Updated 7 years 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
• nimbo-sh / nimbo
  Run compute jobs on AWS as if you were running them locally.
  ☆125Updated 3 years ago
• catid / dataloader
  High-performance tokenized language data-loader for Python C++ extension
  ☆13Updated last year
• NTT123 / pax
  A stateful pytree library for training neural networks.
  ☆22Updated 2 weeks ago
• crowsonkb / LDLM
  Latent Diffusion Language Models
  ☆69Updated last year
• alvarobartt / safejax
  Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`
  ☆45Updated last year
• mhlr / awesome-jax
  List of awesome JAX resources
  ☆13Updated 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
• tmbdev-archive / tarproc
  Utilities for sequential processing of tar files.
  ☆24Updated 3 years ago
• sytelus / pcprep
  Various handy scripts to quickly setup new Linux and Windows sandboxes, containers and WSL.
  ☆40Updated this week
• jax-ml / australis
  ☆28Updated 2 years ago
• dcato98 / fastsparse
  Fastai+PyTorch implementation of sparse model training methods (SET, SNFS, RigL) + customize-your-own.
  ☆10Updated 2 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 4 years ago
• lucidrains / holodeck-pytorch
  Implementation of a holodeck, written in Pytorch
  ☆18Updated last year