Rohith-Rongali's Stars
pytorch-labs/gpt-fast
Simple and efficient pytorch-native transformer text generation in <1000 LOC of python.
benedekrozemberczki/awesome-decision-tree-papers
A collection of research papers on decision, classification and regression trees with implementations.
googlecolab/colabtools
Python libraries for Google Colaboratory
KellerJordan/modded-nanogpt
NanoGPT (124M) in 5 minutes
autodiff/autodiff
automatic differentiation made easier for C++
SamLynnEvans/Transformer
Transformer seq2seq model, program that can build a language translator from parallel corpus
dmarx/anthology-of-modern-ml
Collection of important articles to be treated as a textbook
gpu-mode/awesomeMLSys
An ML Systems Onboarding list
patrick-kidger/lineax
Linear solvers in JAX and Equinox. https://docs.kidger.site/lineax
facebookresearch/luckmatters
Understanding Training Dynamics of Deep ReLU Networks
lmarena/copilot-arena
revalo/iceberg
A compositional diagramming and animation library as an eDSL in Python
google-deepmind/nanodo
f0uriest/interpax
Interpolation and function approximation with JAX
borchero/pycave
Traditional Machine Learning Models for Large-Scale Datasets in PyTorch.
gregorbachmann/Next-Token-Failures
fbach2000/Learning_Theory_from_First_Principles
Python & Matlab code for the figures from the book "Learning Theory from First Principles" by Francis Bach
probabilisticai/nordic-probai-2024
Materials of the Nordic Probabilistic AI School 2024.
amack315/unsupervised-steering-vectors
srush/Tensor-Puzzles-Penzai
chengxiang/LinearTransformer
Pytorch code for experiments on Linear Transformers
dyunis/spectral_dynamics
Code for "Approaching Deep Learning through the Spectral Dynamics of Weights"
pcsl-epfl/hierarchy-learning
Learning a hierarchical model of data with neural networks.
FlorentinGuth/Rainbow
Code to reproduce experiments in "A Rainbow in Deep Network Black Boxes".
nmallinar/rfm-grokking
Grokking modular arithmetic with recursive feature machines
pcsl-epfl/regressionsphere
Train a neural network in feature and lazy regimes on a regression task defined on the hyper-sphere.
Tsili42/parity-nn
aaronpmishkin/experiment_utils
Utilities for running optimization and machine learning experiments.
rajatvd/gifify
rajatvd/sorcerun
Tool to help manage and log computational experiments.