Popular repositories Loading

1. jax jax Public

   Forked from jax-ml/jax

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

   Python

2. jax-llm-examples jax-llm-examples Public

   Forked from jax-ml/jax-llm-examples

   Minimal yet performant LLM examples in pure JAX

   Python

3. jax-triton jax-triton Public

   Forked from jax-ml/jax-triton

   jax-triton contains integrations between JAX and OpenAI Triton

   Python

4. examples examples Public

   Forked from pytorch/examples

   A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.

   Python

5. pytorch pytorch Public

   Forked from pytorch/pytorch

   Tensors and Dynamic neural networks in Python with strong GPU acceleration

   Python

6. optax optax Public

   Forked from google-deepmind/optax

   Optax is a gradient processing and optimization library for JAX.

   Python