Popular repositories Loading

1. pytorch pytorch Public

   Forked from pytorch/pytorch

   Tensors and Dynamic neural networks in Python with strong GPU acceleration

   Python

2. pytorch-xla pytorch-xla Public

   Forked from pytorch/xla

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

   C++

3. flax flax Public

   Forked from google/flax

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

   Jupyter Notebook

4. axlearn axlearn Public

   Forked from apple/axlearn

   An Extensible Deep Learning Library

   Python

5. openxla-shardy openxla-shardy Public

   Forked from openxla/shardy

   MLIR-based partitioning system

   C++

6. openxla-xla openxla-xla Public

   Forked from openxla/xla

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

   C++