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 4 months ago
• che-shr-cat / JAX-in-Action
  Notebooks for the "JAX in Action" book
  ☆122Updated 5 months ago
• stanford-crfm / haliax
  Named Tensors for Legible Deep Learning in JAX
  ☆154Updated last week
• misko / human_descent
  ☆35Updated this week
• wilson-labs / cola
  Compositional Linear Algebra
  ☆449Updated last week
• srush / Autodiff-Puzzles
  ☆394Updated last month
• normal-computing / posteriors
  Uncertainty quantification with PyTorch
  ☆333Updated last month
• modula-systems / modula
  🧱 Modula software package
  ☆130Updated 3 weeks ago
• google-deepmind / nanodo
  ☆199Updated 5 months ago
• Sohl-Dickstein / fractal
  The boundary of neural network trainability is fractal
  ☆164Updated 10 months ago
• probml / ssm-book
  Interactive textbook on state-space models
  ☆178Updated 10 months ago
• HMUNACHI / nanodl
  A Jax-based library for designing and training transformer models from scratch.
  ☆278Updated 3 months ago
• normal-computing / thermox
  Exact OU processes with JAX
  ☆38Updated 2 months ago
• patrick-kidger / quax
  Multiple dispatch over abstract array types in JAX.
  ☆112Updated last month
• xjdr-alt / simple_transformer
  Simple Transformer in Jax
  ☆120Updated 5 months ago
• Artur-Galstyan / statedict2pytree
  ☆40Updated 3 weeks ago
• xl0 / lovely-jax
  JAX Arrays for human consumption
  ☆89Updated last year
• cgarciae / nnx
  Neural Networks for JAX
  ☆83Updated 2 months ago
• itsdaniele / jeometric
  Graph neural networks in JAX.
  ☆67Updated 5 months ago
• samuela / torch2jax
  Run PyTorch in JAX. 🤝
  ☆206Updated 3 weeks ago
• cgarciae / einop
  ☆58Updated 2 years ago
• greydanus / mnist1d
  A 1D analogue of the MNIST dataset for measuring spatial biases and answering Science of Deep Learning questions.
  ☆199Updated 2 months ago
• glassroom / heinsen_sequence
  Code implementing "Efficient Parallelization of a Ubiquitious Sequential Computation" (Heinsen, 2023)
  ☆88Updated last week
• bgavran / Category_Theory_Resources
  List of resources for learning Category Theory
  ☆232Updated 3 weeks ago
• patrick-kidger / optimistix
  Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/
  ☆349Updated 3 weeks ago
• google-deepmind / uncertain_ground_truth
  Dermatology ddx dataset, Jax implementations of Monte Carlo conformal prediction, plausibility regions and statistical annotation aggrega…
  ☆543Updated 8 months ago
• dshah3 / GPU-Puzzles
  Solve puzzles. Learn CUDA.
  ☆60Updated last year
• nocotan / awesome-information-geometry
  About A collection of AWESOME things about information geometry Topics
  ☆150Updated 5 months ago
• diffprog / code
  Code for the book "The Elements of Differentiable Programming".
  ☆68Updated 3 months ago
• google-deepmind / PGMax
  Loopy belief propagation for factor graphs on discrete variables in JAX
  ☆133Updated 2 months ago