Popular repositories Loading

2. Forked from pytorch/pytorch

   Tensors and Dynamic neural networks in Python with strong GPU acceleration

   C++

3. Forked from Lightning-AI/pytorch-lightning

   The lightweight PyTorch wrapper for high-performance AI research. Scale your models, not the boilerplate.

   Python

4. jax jax Public

   Forked from jax-ml/jax

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

   Python

5. Forked from numba/numba

   NumPy aware dynamic Python compiler using LLVM

   Python

6. Forked from tensor-compiler/taco

   The Tensor Algebra Compiler (taco) computes sparse tensor expressions on CPUs and GPUs

   C++