Popular repositories Loading

1. tunix tunix Public

   Forked from google/tunix

   A JAX-native LLM Post-Training Library

   Python

2. sglang-jax sglang-jax Public

   Forked from sgl-project/sglang-jax

   JAX backend for SGL

   Python

3. simply simply Public

   Forked from google-deepmind/simply

   Minimal and scalable research codebase in JAX, designed for rapid iteration on frontier research in LLM and other autoregressive models.

   Python

4. tpu-inference tpu-inference Public

   Forked from vllm-project/tpu-inference

   TPU inference for vLLM, with unified JAX and PyTorch support.

   Python