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"
  ☆106Updated 4 years ago
• ethancaballero / broken_neural_scaling_laws
  Code Release for "Broken Neural Scaling Laws" (BNSL) paper
  ☆58Updated last year
• google / spectral-density
  Hessian spectral density estimation in TF and Jax
  ☆123Updated 4 years ago
• g-benton / loss-surface-simplexes
  ☆99Updated 3 years ago
• rgrosse / csc2541_examples
  ☆29Updated 4 years ago
• aks2203 / deep-thinking
  A centralized place for deep thinking code and experiments
  ☆83Updated last year
• locuslab / edge-of-stability
  ☆67Updated 5 months ago
• aks2203 / easy-to-hard
  Official repository for the paper "Can You Learn an Algorithm? Generalizing from Easy to Hard Problems with Recurrent Networks"
  ☆59Updated 3 years ago
• Sea-Snell / grokking
  unofficial re-implementation of "Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasets"
  ☆78Updated 2 years ago
• srush / torch-golf
  Silly twitter torch implementations.
  ☆46Updated 2 years ago
• nitarshan / robust-generalization-measures
  Official code for "In Search of Robust Measures of Generalization" (NeurIPS 2020)
  ☆28Updated 4 years ago
• lixilinx / psgd_torch
  Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…
  ☆174Updated this week
• teddykoker / grokking
  PyTorch implementation of "Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasets"
  ☆36Updated 3 years ago
• Ping-C / optimizer
  This repository includes code to reproduce the tables in "Loss Landscapes are All You Need: Neural Network Generalization Can Be Explaine…
  ☆36Updated 2 years ago
• 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 2 years ago
• google-research / jax-influence
  ☆60Updated 3 years ago
• js-d / sim_metric
  ☆35Updated last year
• mechanistic-interpretability-grokking / progress-measures-paper
  ☆64Updated 2 years ago
• akbir / deq-jax
  [NeurIPS'19] Deep Equilibrium Models Jax Implementation
  ☆39Updated 4 years ago
• bobby-he / Neural_Tangent_Kernel
  ☆66Updated 6 years ago
• google-deepmind / eigengame
  Open source code for EigenGame.
  ☆30Updated last year
• shikaiqiu / compute-better-spent
  ☆52Updated 7 months ago
• n2cholas / jax-resnet
  Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).
  ☆109Updated 2 years ago
• alecwangcq / KFAC-Pytorch
  Pytorch implementation of KFAC and E-KFAC (Natural Gradient).
  ☆132Updated 5 years ago
• dylandoblar / noether-networks
  Meta-learning inductive biases in the form of useful conserved quantities.
  ☆37Updated 2 years ago
• thudzj / ELLA
  Code for Accelerated Linearized Laplace Approximation for Bayesian Deep Learning (ELLA, NeurIPS 22')
  ☆16Updated 2 years ago
• ganguli-lab / degrees-of-freedom
  ☆37Updated 3 years ago
• HazyResearch / structured-nets
  Structured matrices for compressing neural networks
  ☆66Updated last year
• f-t-s / CGD
  This repository contains the Julia code for the paper "Competitive Gradient Descent"
  ☆24Updated 5 years ago
• jenkspt / gpt-jax
  Jax/Flax rewrite of Karpathy's nanoGPT
  ☆57Updated 2 years ago