Alternatives and similar repositories for gradML

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

• patrick-kidger / typst_pyimage
  Typst extension, adding support for generating figures using inline Python code
  ☆104Updated last month
• saforem2 / ezpz
  Train across all your devices, ezpz 🍋
  ☆24Updated last week
• paganpasta / eqxvision
  A Python package of computer vision models for the Equinox ecosystem.
  ☆109Updated last year
• cgarciae / nnx
  Neural Networks for JAX
  ☆84Updated 11 months ago
• bhosmer / mm
  ☆136Updated 10 months ago
• cgarciae / ciclo
  A functional training loops library for JAX
  ☆88Updated last year
• ravikoka / qgraph
  Simulate Anderson localization and tight binding on graphs.
  ☆14Updated last year
• saforem2 / ambivalent
  Minimal, beautiful (+ highly-customizable) styles for Matplotlib.
  ☆19Updated last month
• mcbal / spin-model-transformers
  Physics-inspired transformer modules based on mean-field dynamics of vector-spin models in JAX
  ☆41Updated last year
• patrick-kidger / patdb
  A snappy + easy + pretty TUI debugger for Python.
  ☆66Updated 2 months ago
• google-deepmind / tf2jax
  ☆115Updated 2 weeks ago
• argonne-lcf / Megatron-DeepSpeed
  Ongoing research training transformer language models at scale, including: BERT & GPT-2
  ☆16Updated last week
• probml / JSL
  Jax SSM Library
  ☆49Updated 2 years ago
• TysonRayJones / PythonTools
  Tools I've written in Python which are useful/necessary for my research
  ☆10Updated 7 years ago
• ludwigwinkler / JaxLightning
  Running Jax in PyTorch Lightning
  ☆113Updated 9 months ago
• xl0 / lovely-jax
  JAX Arrays for human consumption
  ☆106Updated 2 months ago
• RickDW / manim-revealjs
  Plugins for using Manim animations in Reveal.js slide decks.
  ☆60Updated 3 years ago
• yobibyte / compressor
  Because we don't have enough time to read everything
  ☆89Updated 11 months ago
• normal-computing / thermox
  Exact OU processes with JAX
  ☆53Updated 6 months ago
• mcbal / afem
  Implementation of approximate free-energy minimization in PyTorch
  ☆20Updated 3 years ago
• metaopt / optree
  OpTree: Optimized PyTree Utilities
  ☆192Updated last week
• geohot / dumbrl
  Can RL solve simple problems?
  ☆54Updated last year
• kabouzeid / turm
  TUI for the Slurm Workload Manager
  ☆208Updated 3 weeks ago
• jax-ml / jax-ai-stack
  ☆215Updated this week
• stfc-sciml / sciml-bench
  SciML Benchmarking Suite for AI for Science
  ☆42Updated last year
• yobibyte / report
  Because we don't want a jupyter notebook mess...
  ☆61Updated 3 months ago
• deepchem / jaxchem
  JAXChem is a JAX-based deep learning library for complex and versatile chemical modeling
  ☆80Updated 5 years ago
• NVIDIA / JAX-Toolbox
  JAX-Toolbox
  ☆337Updated this week
• j-towns / scanagram
  Tidy autoregressive inference in JAX
  ☆14Updated 3 weeks ago
• profConradi / ALife
  Artificial Life simulations
  ☆53Updated 6 months ago