Popular repositories Loading

1. maxtext maxtext Public

   Forked from AI-Hypercomputer/maxtext

   A simple, performant and scalable Jax LLM!

   Python 1

2. jax jax Public

   Forked from jax-ml/jax

   Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

   Python

3. xla xla Public

   Forked from openxla/xla

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

   C++

4. TransformerEngine TransformerEngine Public

   Forked from NVIDIA/TransformerEngine

   A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit floating point (FP8) precision on Hopper and Ada GPUs, to provide better performance with lower memory utilizatio…

   Python

5. gemma gemma Public

   Forked from google-deepmind/gemma

   Open weights LLM from Google DeepMind.

   Jupyter Notebook