Similar repositories to bhavnicksm/vanilla-transformer-jax:
Bhavnicksm/vanilla-transformer-jax
github
similar
bhavnicksm/vanilla-transformer-jax
github
similar
mukul54/inter-iit-bridgei2i
github
similar
SebastinSanty/minimal-research-theme
github
similar
deepmind/jax_verify
github
similar
google/learned_optimization
github
similar
srush/annotated-s4
github
similar
wan2land/unchartedwater2
github
similar
wan2land/unchartedwaters2
github
similar
zhijing-jin/nlp-phd-global-equality
github
similar
lucidrains/RETRO-pytorch
github
similar
n2cholas/awesome-jax
github
similar
deepmind/tree
github
similar
google-research/deeplab2
github
similar
deepmind/optax
github
similar
deepmind/educational
github
similar
tensorflow/mesh
github
similar
google/flax
github
similar
deepmind/dm-haiku
github
similar
louisfb01/best_AI_papers_2021
github
similar
deepmind/acme
github
similar
deepmind/graph_nets
github
similar
py-why/dowhy
github
similar
activeloopai/Hub
github
similar
google-research/vision_transformer
github
similar
stanfordnlp/stanza
github
similar
arogozhnikov/einops
github
similar
lucidrains/DALLE2-pytorch
github
similar
deepmind/pysc2
github
similar
microsoft/DeepSpeed
github
similar
NVIDIA/DeepLearningExamples
github
similar
networkx/networkx
github
similar
graykode/nlp-tutorial
github
similar
labmlai/annotated_deep_learning_paper_implementations
github
similar
openai/baselines
github
similar
google/jax
github
similar
yunjey/pytorch-tutorial
github
similar
numpy/numpy
github
similar
eugeneyan/applied-ml
github
similar
google-research/google-research
github
similar