Popular repositories Loading

1. pytorch-xla pytorch-xla Public

   Forked from pytorch/xla

   Enabling PyTorch on XLA Devices (e.g. Google TPU)

   Python

2. axlearn axlearn Public

   Forked from apple/axlearn

   An Extensible Deep Learning Library

   Python

3. flax flax Public

   Forked from google/flax

   Flax is a neural network library for JAX that is designed for flexibility.

   Jupyter Notebook

4. pytorch pytorch Public

   Forked from pytorch/pytorch

   Tensors and Dynamic neural networks in Python with strong GPU acceleration

   Python

5. openxla-xla openxla-xla Public

   Forked from openxla/xla

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

   C++

6. jax jax Public

   Forked from jax-ml/jax

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

   Python