Similar repositories to n2cholas/progan-flax:

• n2cholas/progan-flax github similar
• danistefanovic/build-your-own-x github similar
• hanxiao/bert-as-service github similar
• yandex/YaLM-100B github similar
• asmith26/jax_toolkit github similar
• xplip/pixel github similar
• nestordemeure/jochastic github similar
• wandb/wandb github similar
• matthias-wright/jax-fid github similar
• lucidrains/local-attention-flax github similar
• omron-sinicx/jaxmapp github similar
• AranKomat/Alpha-Transformer github similar
• Sea-Snell/MLLibCpp github similar
• vballoli/vit-flax github similar
• rolandgvc/flaxvision github similar
• google/praxis github similar
• deepmind/jax github similar
• sayakpaul/BiT-jax2tf github similar
• sayakpaul/MLPMixer-jax2tf github similar
• myagues/flax_nerf github similar
• sayakpaul/ViT-jax2tf github similar
• thesephist/stream github similar
• gcucurull/maml_flax github similar
• 8bitmp3/JAX-Flax-Tutorial-Image-Classification-with-Linen github similar
• AnirudhDagar/MessagePassing_for_GNNs github similar
• distillpub/post--understanding-rl-vision github similar
• AranKomat/Metroplex github similar
• MilesCranmer/gnn_resource_allocation github similar
• lucidrains/mlp-gpt-jax github similar
• sayakpaul/Denoised-Smoothing-TF github similar
• zchuning/latco github similar
• umangjpatel/kerax github similar
• google/init2winit github similar
• iclr-blog-track/iclr-blog-track.github.io github similar
• shawwn/ml-notes github similar
• BY571/Munchausen-RL github similar
• google/paxml github similar
• ericjang/pt-jax github similar
• srihari-humbarwadi/retinanet-tensorflow2.x github similar
• mcarilli/mixed_precision_references github similar