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"
  ☆105Updated 4 years ago
• google / spectral-density
  Hessian spectral density estimation in TF and Jax
  ☆122Updated 4 years ago
• ethancaballero / broken_neural_scaling_laws
  Code Release for "Broken Neural Scaling Laws" (BNSL) paper
  ☆58Updated last year
• teddykoker / grokking
  PyTorch implementation of "Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasets"
  ☆36Updated 3 years ago
• locuslab / edge-of-stability
  ☆65Updated 3 months ago
• aks2203 / deep-thinking
  A centralized place for deep thinking code and experiments
  ☆82Updated last year
• Sea-Snell / grokking
  unofficial re-implementation of "Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasets"
  ☆75Updated 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"
  ☆60Updated 3 years ago
• g-benton / loss-surface-simplexes
  ☆98Updated 3 years ago
• n2cholas / jax-resnet
  Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).
  ☆108Updated 2 years ago
• wattenberg / superposition
  Code associated to papers on superposition (in ML interpretability)
  ☆28Updated 2 years ago
• bobby-he / Neural_Tangent_Kernel
  ☆67Updated 5 years ago
• ruosongwang / CNTK
  Convolutional Neural Tangent Kernel
  ☆109Updated 5 years ago
• lixilinx / psgd_torch
  Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…
  ☆169Updated 2 months ago
• facebookresearch / projUNN
  Fast training of unitary deep network layers from low-rank updates
  ☆29Updated 2 years ago
• google-research / jax-influence
  ☆60Updated 3 years ago
• krandiash / quinine
  A library to create and manage configuration files, especially for machine learning projects.
  ☆77Updated 2 years ago
• danielmamay / grokking
  Implementation of OpenAI's 'Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasets' paper.
  ☆35Updated last year
• damaru2 / ntk
  NTK reading group
  ☆88Updated 5 years ago
• rgrosse / csc2541_examples
  ☆29Updated 4 years ago
• p-lambda / incontext-learning
  Experiments and code to generate the GINC small-scale in-context learning dataset from "An Explanation for In-context Learning as Implici…
  ☆103Updated last year
• vvvm23 / mamba-jax
  Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX
  ☆83Updated last year
• js-d / sim_metric
  ☆35Updated last year
• shikaiqiu / compute-better-spent
  ☆52Updated 5 months ago
• google-deepmind / jmp
  JMP is a Mixed Precision library for JAX.
  ☆193Updated last month
• google-deepmind / eigengame
  Open source code for EigenGame.
  ☆30Updated last year
• alecwangcq / KFAC-Pytorch
  Pytorch implementation of KFAC and E-KFAC (Natural Gradient).
  ☆130Updated 5 years ago
• jenkspt / gpt-jax
  Jax/Flax rewrite of Karpathy's nanoGPT
  ☆57Updated 2 years ago
• f-t-s / CGD
  This repository contains the Julia code for the paper "Competitive Gradient Descent"
  ☆23Updated 5 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