Similar repositories to zou3519/functorch:
pytorch/functorch
github
similar
danistefanovic/build-your-own-x
github
similar
pytorch/data
github
similar
hanxiao/bert-as-service
github
similar
pytorch/torchdynamo
github
similar
facebookresearch/fairscale
github
similar
deepmind/dm-haiku
github
similar
patrick-kidger/torchtyping
github
similar
pandas-profiling/pandas-profiling
github
similar
patrick-kidger/equinox
github
similar
google/flax
github
similar
lab-ml/annotated_deep_learning_paper_implementations
github
similar
arogozhnikov/einops
github
similar
facebookresearch/xformers
github
similar
patrick-kidger/diffrax
github
similar
google/jaxopt
github
similar
deepmind/optax
github
similar
tuvtran/project-based-learning
github
similar
libffcv/ffcv
github
similar
n2cholas/awesome-jax
github
similar
facebookresearch/higher
github
similar
huggingface/accelerate
github
similar
getkeops/keops
github
similar
PyTorchLightning/lightning-bolts
github
similar
google/neural-tangents
github
similar
pytorch/xla
github
similar
jettify/pytorch-optimizer
github
similar
deepmind/chex
github
similar
rentruewang/koila
github
similar
cgnorthcutt/cleanlab
github
similar
openai/triton
github
similar
cvxgrp/cvxpylayers
github
similar
lucidrains/x-transformers
github
similar
mosaicml/composer
github
similar
mlfoundations/open_clip
github
similar
teddykoker/torchsort
github
similar
facebookresearch/bitsandbytes
github
similar
google/brax
github
similar
google-research/torchsde
github
similar
tootsuite/mastodon
github
similar