Pinned Loading

1. jax jax Public

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

   Python 32.7k 3.1k

2. jax-triton jax-triton Public

   jax-triton contains integrations between JAX and OpenAI Triton

   Python 405 49

3. ml_dtypes ml_dtypes Public

   A stand-alone implementation of several NumPy dtype extensions used in machine learning.

   C++ 277 41