Fast and memory-efficient exact attention
Deep learning library
Models and examples built with TensorFlow
Convert TensorFlow, Keras, Tensorflow.js and Tflite models to ONNX
Deep learning driven jazz generation using Keras & Theano
Streamline your ML workflow
Hummingbird compiles trained ML models into tensor computation
An open source implementation of CLIP
Practice implementing softmax, attention, GPT-2 and more
Fast, flexible and easy to use probabilistic modelling in Python
Extensible, parallel implementations of t-SNE
Implementation of Denoising Diffusion Probabilistic Model in Pytorch
Python package for AutoML on Tabular Data with Feature Engineering
JAX-based neural network library
An easy-to-use & supercharged open-source experiment tracker
DoWhy is a Python library for causal inference
AIMET is a library that provides advanced quantization and compression
Usable Implementation of "Bootstrap Your Own Latent" self-supervised
Trainable models and NN optimization tools
JAX-based neural network library
A refreshing functional take on deep learning
Library for training machine learning models with privacy for data
It's possible for machines to become self-aware.
Useful extra functionality for TensorFlow 2.x maintained by SIG-addons
Differentiable SDE solvers with GPU support and efficient sensitivity