Flax
Showing 12 of 12 projects
Official JAX/Flax implementation of Vision Transformer (ViT) and MLP-Mixer for image recognition, with pre-trained models.
A high-quality modern 3D game engine written in C++ and C# with a focus on fast workflow and stunning graphics.
A JAX library for rapid prototyping of large-scale attention-based vision models across images, video, audio, and multimodal data.
A JAX/Flax-based framework for easy and scalable pre-training, fine-tuning, evaluation, and serving of large language models.
A collection of tutorials and resources to help developers learn JAX, Flax, and Haiku for machine learning.
JAX (Flax) implementations of reinforcement learning algorithms for continuous action spaces, designed for research.
A low-level Gaussian process framework in JAX and Flax, designed for maximum flexibility and close alignment with mathematical notation.
A collection of pretrained deep learning models (StyleGAN2, GPT2, VGG, ResNet) for the Jax/Flax ecosystem.
Unofficial JAX/Flax implementations of deep learning research papers for vision transformers and other architectures.
Flax implementations and pretrained checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX.
Official JAX implementation of XMC-GAN for text-to-image generation using cross-modal contrastive learning.
A JAX + Flax implementation of physics-inspired graph neural networks for solving combinatorial optimization problems like Max-Cut and Maximum Independent Set.
Related Tags
Found a gem we're missing?
Open-Awesome is built by the community, for the community. Submit a project, suggest an awesome list, or help improve the catalog on GitHub.