PyTorch, TensorFlow, JAX and NumPy — all of them natively using the same code
MIT License
A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit floating poin...
Just you average function (there is nothing going on)
Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https...
Deep Learning and Reinforcement Learning Library for Scientists and Engineers
Unofficial implementation of Tensorial Radiance Fields (Chen & Xu ‘22)
Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/
A Python toolbox to create adversarial examples that fool neural networks in PyTorch, TensorFlow,...
Intuitive scientific computing with dimension types for Jax, PyTorch, TensorFlow & NumPy
TensorLy: Tensor Learning in Python.
The Unified AI Framework
The goal of this library is to generate more helpful exception messages for matrix algebra expres...
A suite of benchmarks for CPU and GPU performance of the most popular high-performance libraries ...
🚀 A simple way to launch, train, and use PyTorch models on almost any device and distributed conf...
In this repository, I will share some useful notes and references about deploying deep learning-b...
Deep Learning for humans