flax topic

EfficientNet, MobileNetV3, MobileNetV2, MixNet, etc in JAX w/ Flax Linen and Objax

rwightman

Tevatron - Unified Document Retrieval Toolkit across Scale, Language, and Modality. Demo in SIGIR 2023, SIGIR 2025.

texttron

Long Range Arena for Benchmarking Efficient Transformers

google-research

Original Implementation of Prompt Tuning from Lester, et al, 2021

google-research

Run Effective Large Batch Contrastive Learning Beyond GPU/TPU Memory Constraint

luyug

general purpose programming language, in the vein of C++

flax-lang

CLU lets you write beautiful training loops in JAX.

google

Unofficial JAX implementations of deep learning research papers

DarshanDeshpande

KoCLIP: Korean port of OpenAI CLIP, in Flax

jaketae

Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).

n2cholas