Similar repositories to google/flax:

• google/flax github similar
• deepmind/dm-haiku github similar
• google/jax github similar
• danistefanovic/build-your-own-x github similar
• huggingface/diffusers github similar
• deepmind/optax github similar
• arogozhnikov/einops github similar
• mosaicml/composer github similar
• patrick-kidger/equinox github similar
• huggingface/accelerate github similar
• deepmind/rlax github similar
• pytorch/functorch github similar
• lucidrains/DALLE2-pytorch github similar
• HIPS/autograd github similar
• microsoft/DeepSpeed github similar
• gradio-app/gradio github similar
• openai/triton github similar
• cupy/cupy github similar
• hanxiao/bert-as-service github similar
• google-research/scenic github similar
• facebookresearch/metaseq github similar
• libffcv/ffcv github similar
• lucidrains/imagen-pytorch github similar
• tuvtran/project-based-learning github similar
• deepmind/jraph github similar
• google/trax github similar
• pyro-ppl/numpyro github similar
• CompVis/latent-diffusion github similar
• NVIDIA/apex github similar
• facebookresearch/hydra github similar
• openai/CLIP github similar
• deepmind/mujoco github similar
• optuna/optuna github similar
• google-research/vision_transformer github similar
• lucidrains/denoising-diffusion-pytorch github similar
• unifyai/ivy github similar
• DLR-RM/stable-baselines3 github similar
• lucidrains/x-transformers github similar
• microsoft/unilm github similar
• pytorch/captum github similar