Continuous-Time/State/Action Fitted Value Iteration via Hamilton-Jacobi-Bellman (HJB)
JAX-based Model Explanation and Interpretation Library
A pathway and collection of resources to learning Jax from beginning to advance.
Training GPT-2 on FineWeb-Edu in JAX/Flax
An app with FastAPI, Docker, transformers, JAX/Flax for performing image segmentation with PaliGemma 2 mix
A diffusion transformer implementation in Flax
Another site generator Melt.
FlaUI wrapper(UIA3 only), adding Java and OpenCV feature
Image classification in JAX with ViT, resnet, cifar10, cifar100, imagenette, and imagenet
A compilation of most State-of-the-Art Attention Mechanisms: MHSA, MQA, GQA, and MLA, implemented in JAX
This repository implements sentiment classifier with Google Jax using Bert transformer as backbone. It also shows model checkpointing and loading for inference.
Add a description, image, and links to the flax topic page so that developers can more easily learn about it.
To associate your repository with the flax topic, visit your repo's landing page and select "manage topics."