A library for easy and efficient manipulation of tensor networks.
APACHE-2.0 License
Deep Learning and Reinforcement Learning Library for Scientists and Engineers
Jittor is a high-performance deep learning framework based on JIT compiling and meta-operators.
GPU Accelerated t-SNE for CUDA with Python bindings
Intuitive scientific computing with dimension types for Jax, PyTorch, TensorFlow & NumPy
Hyper optimized contraction trees for large tensor networks and einsums
TensorLayerX: A Unified Deep Learning and Reinforcement Learning Framework for All Hardwares, Bac...
Implementation of E(n)-Equivariant Graph Neural Networks, in Pytorch
A JAX research toolkit for building, editing, and visualizing neural networks.
TensorLy: Tensor Learning in Python.
The goal of this library is to generate more helpful exception messages for matrix algebra expres...
A Graph Neural Network Library in Jax
Neural Network Libraries