Pinned Loading

1. jax-ml/jax jax-ml/jax Public

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

   Python 33.8k 3.2k

2. google/brax google/brax Public

   Massively parallel rigidbody physics simulation on accelerator hardware.

   Jupyter Notebook 2.9k 316

3. google/BIG-bench google/BIG-bench Public

   Beyond the Imitation Game collaborative benchmark for measuring and extrapolating the capabilities of language models

   Python 3.1k 614

4. google/learned_optimization google/learned_optimization Public

   Python 785 70