Alternatives and similar repositories for TP4

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

• thegregyang / NTK4A
  Code for the paper: "Tensor Programs II: Neural Tangent Kernel for Any Architecture"
  ☆109Updated 5 years ago
• google / spectral-density
  Hessian spectral density estimation in TF and Jax
  ☆125Updated 5 years ago
• ethancaballero / broken_neural_scaling_laws
  Code Release for "Broken Neural Scaling Laws" (BNSL) paper
  ☆59Updated 2 years ago
• aks2203 / easy-to-hard
  Official repository for the paper "Can You Learn an Algorithm? Generalizing from Easy to Hard Problems with Recurrent Networks"
  ☆61Updated 3 years ago
• SirRob1997 / Crowded-Valley---Results
  This repository contains the results for the paper: "Descending through a Crowded Valley - Benchmarking Deep Learning Optimizers"
  ☆184Updated 4 years ago
• google-deepmind / neural_networks_chomsky_hierarchy
  Neural Networks and the Chomsky Hierarchy
  ☆212Updated last year
• HEmile / storchastic
  Stochastic Automatic Differentiation library for PyTorch.
  ☆208Updated last year
• dylandoblar / noether-networks
  Meta-learning inductive biases in the form of useful conserved quantities.
  ☆39Updated 3 years ago
• g-benton / loss-surface-simplexes
  ☆100Updated 4 years ago
• Felix-Petersen / algovision
  Differentiable Algorithms and Algorithmic Supervision.
  ☆116Updated 2 years ago
• lixilinx / psgd_torch
  Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…
  ☆190Updated last month
• srush / mamba-scans
  Blog post
  ☆17Updated last year
• google-deepmind / tf2jax
  ☆120Updated 2 weeks ago
• n2cholas / jax-resnet
  Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).
  ☆119Updated 3 years ago
• teddykoker / grokking
  PyTorch implementation of "Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasets"
  ☆38Updated 4 years ago
• davisyoshida / lorax
  LoRA for arbitrary JAX models and functions
  ☆144Updated last year
• kazukiosawa / asdl
  ASDL: Automatic Second-order Differentiation Library for PyTorch
  ☆191Updated last year
• RobertCsordas / transformer_generalization
  The official repository for our paper "The Devil is in the Detail: Simple Tricks Improve Systematic Generalization of Transformers". We s…
  ☆67Updated 3 years ago
• google-deepmind / jmp
  JMP is a Mixed Precision library for JAX.
  ☆211Updated last year
• aks2203 / deep-thinking
  A centralized place for deep thinking code and experiments
  ☆90Updated 2 years ago
• KindXiaoming / Omnigrok
  Omnigrok: Grokking Beyond Algorithmic Data
  ☆62Updated 2 years ago
• ischlag / fast-weight-transformers
  Official code repository of the paper Linear Transformers Are Secretly Fast Weight Programmers.
  ☆111Updated 4 years ago
• krandiash / quinine
  A library to create and manage configuration files, especially for machine learning projects.
  ☆79Updated 3 years ago
• bremen79 / parameterfree
  Parameter-Free Optimizers for Pytorch
  ☆130Updated last year
• google-research / jaxpruner
  ☆234Updated last year
• lucidrains / PaLM-jax
  Implementation of the specific Transformer architecture from PaLM - Scaling Language Modeling with Pathways - in Jax (Equinox framework)
  ☆190Updated 3 years ago
• google-deepmind / jaxline
  ☆163Updated 2 years ago
• HazyResearch / structured-nets
  Structured matrices for compressing neural networks
  ☆67Updated 2 years ago
• Sea-Snell / JAXSeq
  Train very large language models in Jax.
  ☆210Updated 2 years ago
• gpauloski / kfac-pytorch
  Distributed K-FAC preconditioner for PyTorch
  ☆95Updated this week