Pinned Loading

1. openxla/xla openxla/xla Public

   A machine learning compiler for GPUs, CPUs, and ML accelerators

   C++ 4.6k 979

2. vllm-project/vllm-omni vllm-project/vllm-omni Public

   A framework for efficient model inference with omni-modality models

   Python 7.3k 2k

3. mistral-v0.2-jax mistral-v0.2-jax Public

   JAX implementation of the Mistral 7b v0.2 model

   Python 35 1

4. einshard einshard Public

   Einsum-like high-level array sharding API for JAX

   Python 35 2

5. tpux tpux Public

   A set of Python scripts that makes your experience on TPU better

   Python 56 3

6. tpu-training-example tpu-training-example Public

   Python 16 1