Code accompanying our paper "Feature Learning in Infinite-Width Neural Networks" (https://arxiv.org/abs/2011.14522)
☆61May 11, 2021Updated 5 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:
- Code for the paper: "Tensor Programs II: Neural Tangent Kernel for Any Architecture"☆110Aug 26, 2020Updated 6 years ago
- ☆14Oct 18, 2021Updated 4 years ago
- Visualization of mean field and neural tangent kernel regime☆23Jul 25, 2024Updated 2 years ago
- Convolutional Neural Tangent Kernel☆113Oct 22, 2019Updated 6 years ago
- NTK reading group☆86Nov 14, 2019Updated 6 years ago
- Serverless GPU API endpoints on Runpod - Get Bonus Credits • AdSkip the infrastructure headaches. Auto-scaling, pay-as-you-go, no-ops approach lets you focus on innovating your application.
- ☆15Sep 29, 2022Updated 4 years ago
- Fast and Easy Infinite Neural Networks in Python☆2,385Mar 1, 2024Updated 2 years ago
- Using Huggingface to generate relation expressions☆15Jan 15, 2021Updated 5 years ago
- Tidy autoregressive inference in JAX☆15Sep 1, 2025Updated last year
- Code for SaGe subword tokenizer (EACL 2023)☆28Nov 30, 2024Updated last year
- maximal update parametrization (µP)☆1,756Jul 17, 2024Updated 2 years ago
- ☆11Dec 6, 2020Updated 5 years ago
- An ultra-lightweight JAX implementation of sparse Gaussian processes via pathwise sampling.☆22Mar 31, 2021Updated 5 years ago
- ☆12Oct 8, 2020Updated 6 years ago
- Simple, predictable pricing with DigitalOcean hosting • AdAlways know what you'll pay with monthly caps and flat pricing. Enterprise-grade infrastructure trusted by 600k+ customers.
- This repository hosts the code to port NumPy model weights of BiT-ResNets to TensorFlow SavedModel format.☆14Dec 21, 2021Updated 4 years ago
- Code for steering and monitoring with concepts vectors in LLMs (initial draft)☆33Aug 10, 2025Updated last year
- Repo reproducing experimental results in "Addressing the Topological Defects of Disentanglement"☆24Jul 15, 2022Updated 4 years ago
- Python implementation of paper "AntisymmetricRNN: A Dynamical System View on Recurrent Neural Networks"☆15Aug 2, 2019Updated 7 years ago
- ☆17Feb 4, 2025Updated last year
- This repository contains the code used for the experiments in the paper "Fine-Tuning Enhances Existing Mechanisms: A Case Study on Entity…☆32Oct 27, 2025Updated 11 months ago
- ☆16May 16, 2023Updated 3 years ago
- This is a PyTorch reimplementation of Influence Functions from the ICML2017 best paper: Understanding Black-box Predictions via Influence…☆17Jun 20, 2020Updated 6 years ago
- [NeurIPS 2022] DataMUX: Data Multiplexing for Neural Networks☆60Nov 24, 2022Updated 3 years ago
- Serverless GPU API endpoints on Runpod - Get Bonus Credits • AdSkip the infrastructure headaches. Auto-scaling, pay-as-you-go, no-ops approach lets you focus on innovating your application.
- An ML research codebase built with friends :)☆26Aug 25, 2024Updated 2 years ago
- Experimental version of jxbz/agd implementing support for bias terms, affine parameters, transformers, etc.☆12Jul 30, 2023Updated 3 years ago
- ☆32Apr 4, 2024Updated 2 years ago
- Code for verifying deep neural feature ansatz☆23May 3, 2023Updated 3 years ago
- Simple implementation of muP, based on Spectral Condition for Feature Learning. The implementation is SGD only, dont use it for Adam☆89Jul 28, 2024Updated 2 years ago
- ☆33Apr 12, 2021Updated 5 years ago
- ☆20Feb 28, 2025Updated last year
- A lightweight didactic library of kernel methods using the back-end JAX.☆12Mar 8, 2023Updated 3 years ago
- Easy training for Sparse Linear Autoencoders (https://arxiv.org/abs/2309.08600) with data from TransformerLens models.☆12Nov 1, 2023Updated 2 years ago
- GPUs on demand by Runpod - Special Offer Available • AdRun AI, ML, and HPC workloads on powerful cloud GPUs—without limits or wasted spend. Deploy GPUs in under a minute and pay by the second.
- Code for the Shortformer model, from the ACL 2021 paper by Ofir Press, Noah A. Smith and Mike Lewis.☆145Jul 26, 2021Updated 5 years ago
- (Python, R, C) Fast approximations for the CDF of multivariate normal distributions☆31Jun 12, 2026Updated 4 months ago
- The project of recognizing handwritten digits found in mnist dataset by the use of Neural Networks(Feedforward Fully Connected)☆11Sep 22, 2022Updated 4 years ago
- Effective Attention Sheds Light On Interpretability - Findings of ACL2021☆11May 16, 2021Updated 5 years ago
- ☆110Feb 18, 2026Updated 7 months ago
- incremental symbol learning for natural language understanding☆10Jun 12, 2023Updated 3 years ago
- Implementation of Retrieval-Augmented Denoising Diffusion Probabilistic Models in Pytorch☆66May 5, 2022Updated 4 years ago