Pinned

1. google/jax Public

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

   Python 22.2k 2.1k

2. google/praxis Public

   Python 71 11

3. google/paxml Public

   Python 83 12

4. tensorflow/lingvo Public

   Lingvo

   Python 2.7k 430

5. tensorflow/runtime Public

   A performant and modular runtime for TensorFlow

   MLIR 713 123

6. spmd_pipeline_example.py

   1

   import itertools as it

   2

   3

   import jax

   4

   import jax.numpy as jnp

   5