Alternatives and similar repositories for nanograd

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

• rasmusbergpalm / nanograd
  A miniscule implementation of reverse mode auto-differentiation
  ☆29Updated 3 years ago
• cgarciae / nnx
  Neural Networks for JAX
  ☆84Updated 9 months ago
• cgarciae / einop
  ☆60Updated 3 years ago
• alvarobartt / safejax
  Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`
  ☆45Updated last year
• awf / functional-transformer
  A pure-functional implementation of a machine learning transformer model in Python/JAX
  ☆179Updated last month
• google-deepmind / nanodo
  ☆270Updated 11 months ago
• sytelus / pcprep
  Various handy scripts to quickly setup new Linux and Windows sandboxes, containers and WSL.
  ☆40Updated 2 months ago
• ClashLuke / tpucare
  Automatically take good care of your preemptible TPUs
  ☆36Updated 2 years ago
• borisdayma / clip-jax
  Train vision models using JAX and 🤗 transformers
  ☆98Updated 2 months ago
• google-deepmind / tf2jax
  ☆114Updated 3 weeks ago
• paganpasta / eqxvision
  A Python package of computer vision models for the Equinox ecosystem.
  ☆107Updated 11 months ago
• lixilinx / psgd_torch
  Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…
  ☆177Updated 2 weeks ago
• dvruette / barrel-rec-pytorch
  ☆53Updated last year
• HomebrewML / Olmax
  HomebrewNLP in JAX flavour for maintable TPU-Training
  ☆50Updated last year
• BlackHC / neural_net_checklist
  ☆150Updated 10 months ago
• AakashKumarNain / diffusion_models
  All about the fundamentals and working of Diffusion Models
  ☆158Updated 2 years ago
• Sohl-Dickstein / fractal
  The boundary of neural network trainability is fractal
  ☆208Updated last year
• HomebrewML / HeavyBall
  Efficient optimizers
  ☆220Updated this week
• cgarciae / ciclo
  A functional training loops library for JAX
  ☆88Updated last year
• AakashKumarNain / mistral_jax
  This is a port of Mistral-7B model in JAX
  ☆32Updated 11 months ago
• saurabhaloneai / Llama-3-From-Scratch-In-Pure-Jax
  This repository contain the simple llama3 implementation in pure jax.
  ☆66Updated 4 months ago
• DarshanDeshpande / jax-models
  Unofficial JAX implementations of deep learning research papers
  ☆156Updated 3 years ago
• modula-systems / modula
  🧱 Modula software package
  ☆200Updated 2 months ago
• bhoov / hamux
  Hierarchical Associative Memory User Experience
  ☆100Updated last year
• samuela / torch2jax
  Run PyTorch in JAX. 🤝
  ☆252Updated 4 months ago
• hundredblocks / large-model-parallelism
  Functional local implementations of main model parallelism approaches
  ☆95Updated 2 years ago
• irregular-rhomboid / EAI-Math-Reading-Group
  Resources from the EleutherAI Math Reading Group
  ☆53Updated 3 months ago
• GallagherCommaJack / modulax
  ☆17Updated 10 months ago
• cgarciae / treex
  A Pytree Module system for Deep Learning in JAX
  ☆214Updated 2 years ago
• enajx / HyperNCA
  ☆39Updated 3 years ago