Alternatives and similar repositories for gradML

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

• patrick-kidger / patdb
  A snappy + easy + pretty TUI debugger for Python.
  ☆30Updated this week
• ekzhang / ekzhang.sty
  My personal LaTeX template, with sensible formatting and commands
  ☆16Updated 2 years ago
• RylanSchaeffer / Stanford-AI-Alignment-Double-Descent-Tutorial
  Code for Arxiv Double Descent Demystified: Identifying, Interpreting & Ablating the Sources of a Deep Learning Puzzle
  ☆26Updated last year
• cgarciae / nnx
  Neural Networks for JAX
  ☆84Updated 8 months ago
• probml / JSL
  Jax SSM Library
  ☆49Updated 2 years ago
• alexzhang13 / Triton-Puzzles-Solutions
  Personal solutions to the Triton Puzzles
  ☆18Updated 10 months ago
• cgarciae / einop
  ☆60Updated 3 years ago
• mancusolab / traceax
  Stochastic trace estimation using JAX
  ☆15Updated 2 months ago
• Lightning-Universe / open-bio-ml-workshop
  ☆10Updated 2 years ago
• wattenberg / superposition
  Code associated to papers on superposition (in ML interpretability)
  ☆28Updated 2 years ago
• phcerdan / wolfram_model
  Python Wrappings for exploring Set Substitution Systems (Wolfram Models)
  ☆15Updated 5 years ago
• etimush / ARC_NCA
  Repo for solving arc problems with an Neural Cellular Automata
  ☆15Updated 2 weeks ago
• mcbal / spin-model-transformers
  Physics-inspired transformer modules based on mean-field dynamics of vector-spin models in JAX
  ☆41Updated last year
• paganpasta / eqxvision
  A Python package of computer vision models for the Equinox ecosystem.
  ☆106Updated 10 months ago
• patrick-kidger / esm2quinox
  An implementation of ESM2 in Equinox+JAX
  ☆27Updated this week
• ml-jku / quam
  Quantification of Uncertainty with Adversarial Models
  ☆29Updated last year
• nboyd / joltz
  JAX translation of boltz
  ☆18Updated 2 weeks ago
• google-deepmind / transformer_ngrams
  ☆23Updated 7 months ago
• gata-bio / affine-gaps
  Less-wrong single-file Numba-accelerated Python implementation of Gotoh affine gap penalty extensions for the Needleman–Wunsch, Smith-Wat…
  ☆9Updated 2 months ago
• tychovdo / noethers-razor
  Code for NeurIPS 2024 paper: "Noether's razor: Learning Conserved Quantities" by Tycho F. A. van der Ouderaa, Mark van der Wilk, Pim de H…
  ☆10Updated 7 months ago
• deepchem / jaxchem
  JAXChem is a JAX-based deep learning library for complex and versatile chemical modeling
  ☆79Updated 4 years ago
• fferflo / weightbridge
  Map (deep learning) model weights between different model implementations.
  ☆19Updated 4 months ago
• GallagherCommaJack / modulax
  ☆17Updated 9 months ago
• srush / drop7
  ☆18Updated last year
• Artur-Galstyan / statedict2pytree
  ☆43Updated last week
• HumanCompatibleAI / leela-interp
  Code for "Evidence of Learned Look-Ahead in a Chess-Playing Neural Network"
  ☆22Updated last year
• EricDarve / NLA
  Numerical Linear Algebra Notes CME 302 Stanford
  ☆14Updated 6 months ago
• somacdivad / grinpy
  A NetworkX extension for calculating graph invariants.
  ☆20Updated 2 years ago
• patrick-kidger / quax
  Multiple dispatch over abstract array types in JAX.
  ☆124Updated this week
• whitead / minimalaf
  Minimal AlphaFold
  ☆53Updated 9 months ago