Medical imaging toolkit for deep learning
High-level training, data augmentation, and utilities for Pytorch
JAX-based neural network library
Democratizing Deep-Learning for Drug Discovery, Quantum Chemistry, etc
Making large AI models cheaper, faster and more accessible
The Unified Machine Learning Framework
Open deep learning compiler stack for cpu, gpu, etc.
AI discovers 520000 stable inorganic crystal structures for research
NVIDIA Federated Learning Application Runtime Environment
A Python library for audio data augmentation
PyTorch extensions for fast R&D prototyping and Kaggle farming
Spatiotemporal Signal Processing with Neural Machine Learning Models
Data science on data without acquiring a copy
Python package for AutoML on Tabular Data with Feature Engineering
Open Source Differentiable Computer Vision Library
Fast image augmentation library and an easy-to-use wrapper
A library for deep learning end-to-end dialog systems and chatbots
AIMET is a library that provides advanced quantization and compression
Evaluate and monitor ML models from validation to production
A library for easily evaluating machine learning models and datasets
Message Passing Neural Networks for Molecule Property Prediction
Hummingbird compiles trained ML models into tensor computation
Python implementation of TextRank algorithms
Functional Machine Learning
A library of extension and helper modules for Python's data analysis