Pinned

1. jax-clm-sft jax-clm-sft Public

   Supervised Transformer Fine-Tuning (SFT) with a Causal Language Modelling (CLM) objective.

   Python

2. vit-image-classification vit-image-classification Public

   Fine tune ViT for image classification. Built with JAX/Flax/HuggingFace and others.

   Python 2

3. jax-pjit-examples jax-pjit-examples Public

   Leverage JAX pjit to run models that won't fit on a single accelerator.

   Python

4. vllm-project/vllm vllm-project/vllm Public

   A high-throughput and memory-efficient inference and serving engine for LLMs

   Python 18.8k 2.5k