Alternatives and similar repositories for tensorgrad:

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

• vtabbott / Algebraic-NCD
  A package for defining deep learning models using categorical algebraic expressions.
  ☆58Updated 5 months ago
• che-shr-cat / JAX-in-Action
  Notebooks for the "JAX in Action" book
  ☆125Updated 6 months ago
• misko / human_descent
  ☆36Updated last month
• srush / Autodiff-Puzzles
  ☆413Updated 2 months ago
• wilson-labs / cola
  Compositional Linear Algebra
  ☆456Updated this week
• Sohl-Dickstein / fractal
  The boundary of neural network trainability is fractal
  ☆190Updated 11 months ago
• normal-computing / posteriors
  Uncertainty quantification with PyTorch
  ☆333Updated 2 months ago
• adamnemecek / traceoid.ai
  ☆46Updated last month
• modula-systems / modula
  🧱 Modula software package
  ☆132Updated this week
• stanford-crfm / haliax
  Named Tensors for Legible Deep Learning in JAX
  ☆157Updated last week
• fbach2000 / Learning_Theory_from_First_Principles
  Python & Matlab code for the figures from the book "Learning Theory from First Principles" by Francis Bach
  ☆74Updated 10 months ago
• google-deepmind / nanodo
  ☆201Updated 6 months ago
• HMUNACHI / nanodl
  A Jax-based library for designing and training transformer models from scratch.
  ☆280Updated 4 months ago
• srush / Transformer-Puzzles
  Puzzles for exploring transformers
  ☆331Updated last year
• BlackHC / neural_net_checklist
  ☆149Updated 5 months ago
• francoisfleuret / dlc
  ☆40Updated 3 weeks ago
• greydanus / mnist1d
  A 1D analogue of the MNIST dataset for measuring spatial biases and answering Science of Deep Learning questions.
  ☆204Updated 3 months ago
• xjdr-alt / simple_transformer
  Simple Transformer in Jax
  ☆128Updated 6 months ago
• facebookresearch / optimizers
  For optimization algorithm research and development.
  ☆484Updated this week
• google-deepmind / uncertain_ground_truth
  Dermatology ddx dataset, Jax implementations of Monte Carlo conformal prediction, plausibility regions and statistical annotation aggrega…
  ☆642Updated 9 months ago
• probml / ssm-book
  Interactive textbook on state-space models
  ☆181Updated 11 months ago
• itsdaniele / jeometric
  Graph neural networks in JAX.
  ☆67Updated 7 months ago
• haraschax / nograd
  Gradient descent is cool and all, but what if we could delete it?
  ☆102Updated 3 weeks ago
• patrick-kidger / quax
  Multiple dispatch over abstract array types in JAX.
  ☆113Updated 2 weeks ago
• bgavran / Category_Theory_Resources
  List of resources for learning Category Theory
  ☆238Updated last month
• neoneye / ARC-Interactive-History-Dataset
  The history files when recording human interaction while solving ARC tasks
  ☆96Updated last week
• nocotan / awesome-information-geometry
  About A collection of AWESOME things about information geometry Topics
  ☆151Updated 6 months ago
• kvfrans / jax-diffusion-transformer
  Implementation of Diffusion Transformer (DiT) in JAX
  ☆261Updated 7 months ago
• KindXiaoming / BIMT
  Brain-Inspired Modular Training (BIMT), a method for making neural networks more modular and interpretable.
  ☆163Updated last year
• patrick-kidger / optimistix
  Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/
  ☆363Updated 2 weeks ago