Similar repositories to srush/Tensor-Puzzles:
srush/Tensor-Puzzles
github
similar
srush/GPU-Puzzles
github
similar
danistefanovic/build-your-own-x
github
similar
hanxiao/bert-as-service
github
similar
lab-ml/annotated_deep_learning_paper_implementations
github
similar
yandex/YaLM-100B
github
similar
pandas-profiling/pandas-profiling
github
similar
tuvtran/project-based-learning
github
similar
patrick-kidger/equinox
github
similar
jmtomczak/intro_dgm
github
similar
vopani/jaxton
github
similar
n2cholas/awesome-jax
github
similar
cgnorthcutt/cleanlab
github
similar
patrick-kidger/torchtyping
github
similar
facebookresearch/torchdim
github
similar
HazyResearch/state-spaces
github
similar
clementchadebec/benchmark_VAE
github
similar
pytorch/functorch
github
similar
harvardnlp/pytorch-struct
github
similar
mosaicml/composer
github
similar
srush/streambook
github
similar
facebookresearch/xformers
github
similar
deepmind/xmanager
github
similar
pytorch/data
github
similar
srush/annotated-s4
github
similar
google-research/big_vision
github
similar
microsoft/mup
github
similar
google/jaxopt
github
similar
huggingface/evaluate
github
similar
libffcv/ffcv
github
similar
google-research/t5x
github
similar
google/jaxtyping
github
similar
google/brax
github
similar
minitorch/minitorch
github
similar
deepmind/chex
github
similar
mlfoundations/open_clip
github
similar
pytorch/torchdynamo
github
similar
HazyResearch/flash-attention
github
similar
facebookresearch/higher
github
similar
DiffEqML/torchdyn
github
similar