Code accompanying our paper "Feature Learning in Infinite-Width Neural Networks" (https://arxiv.org/abs/2011.14522)
☆62May 11, 2021Updated 4 years ago
Alternatives and similar repositories for TP4
Users that are interested in TP4 are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- ☆14Oct 18, 2021Updated 4 years ago
- Visualization of mean field and neural tangent kernel regime☆23Jul 25, 2024Updated last year
- Code for the paper: "Tensor Programs II: Neural Tangent Kernel for Any Architecture"☆109Aug 26, 2020Updated 5 years ago
- Neural Tangent Kernel Papers☆122Jan 12, 2025Updated last year
- Tidy autoregressive inference in JAX☆15Sep 1, 2025Updated 8 months ago
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- NTK reading group☆85Nov 14, 2019Updated 6 years ago
- A quick test of SIREN on out-of-sample tasks☆13Jul 4, 2020Updated 5 years ago
- Fast and Easy Infinite Neural Networks in Python☆2,382Mar 1, 2024Updated 2 years ago
- maximal update parametrization (µP)☆1,704Jul 17, 2024Updated last year
- ☆11Dec 6, 2020Updated 5 years ago
- Code for steering and monitoring with concepts vectors in LLMs. https://arxiv.org/abs/2502.03708☆29Aug 10, 2025Updated 8 months ago
- An ultra-lightweight JAX implementation of sparse Gaussian processes via pathwise sampling.☆22Mar 31, 2021Updated 5 years ago
- LMS algorithm for computing edit distance with SIMD optimizations☆11Mar 25, 2024Updated 2 years ago
- Fork of diux-dev/imagenet18☆16Oct 4, 2018Updated 7 years ago
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- ☆12Oct 8, 2020Updated 5 years ago
- This repository hosts the code to port NumPy model weights of BiT-ResNets to TensorFlow SavedModel format.☆14Dec 21, 2021Updated 4 years ago
- ☆16Feb 4, 2025Updated last year
- Repo reproducing experimental results in "Addressing the Topological Defects of Disentanglement"☆23Jul 15, 2022Updated 3 years ago
- Python implementation of paper "AntisymmetricRNN: A Dynamical System View on Recurrent Neural Networks"☆15Aug 2, 2019Updated 6 years ago
- The simplest, fastest repository for training/finetuning medium-sized GPTs.☆192Jan 19, 2026Updated 3 months ago
- [NeurIPS 2022] DataMUX: Data Multiplexing for Neural Networks☆60Nov 24, 2022Updated 3 years ago
- sketching algorithms implemented in chapel and python☆10Jun 8, 2017Updated 8 years ago
- ☆18Feb 28, 2025Updated last year
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- An ML research codebase built with friends :)☆25Aug 25, 2024Updated last year
- SD-based Anifusion☆15Dec 23, 2022Updated 3 years ago
- Experimental version of jxbz/agd implementing support for bias terms, affine parameters, transformers, etc.☆12Jul 30, 2023Updated 2 years ago
- Simple implementation of muP, based on Spectral Condition for Feature Learning. The implementation is SGD only, dont use it for Adam☆87Jul 28, 2024Updated last year
- ☆33Apr 12, 2021Updated 5 years ago
- (Python, R, C) Fast approximations for the CDF of multivariate normal distributions☆31May 9, 2025Updated 11 months ago
- PyTorch-based library for various kinds of representational-similarity analysis☆25Jun 7, 2024Updated last year
- incremental symbol learning for natural language understanding☆10Jun 12, 2023Updated 2 years ago
- Python3 Solutions to LeetCode's "Top 100 Liked Questions", “Top Interview Questions”, and Google’s foobar challenge☆20Apr 18, 2023Updated 3 years ago
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- This repo contains code for the paper: "Can Foundation Models Help Us Achieve Perfect Secrecy?"☆24Feb 9, 2023Updated 3 years ago
- Implementation of Retrieval-Augmented Denoising Diffusion Probabilistic Models in Pytorch☆66May 5, 2022Updated 4 years ago
- The Full Spectrum of Deepnet Hessians at Scale: Dynamics with SGD Training and Sample Size☆19May 19, 2019Updated 6 years ago
- A Kernel-Based View of Language Model Fine-Tuning https://arxiv.org/abs/2210.05643☆78Sep 4, 2023Updated 2 years ago
- Implementation of Vision Transformers in Flax☆18Oct 12, 2020Updated 5 years ago
- A lightweight didactic library of kernel methods using the back-end JAX.☆12Mar 8, 2023Updated 3 years ago
- ☆46Oct 11, 2023Updated 2 years ago