A framework for real-life data science
Decentralized deep learning in PyTorch. Built to train models
A unified, comprehensive and efficient recommendation library
Fast image augmentation library and an easy-to-use wrapper
A unified framework for scalable computing
The unified and scalable ML library for large-scale training
A machine learning library for detecting anomalies in signals
A library for scientific machine learning & physics-informed learning
A library for audio and music analysis, feature extraction
PyTorch extensions for fast R&D prototyping and Kaggle farming
Medical imaging toolkit for deep learning
DeepMind's software stack for physics-based simulation
Probabilistic reasoning and statistical analysis in TensorFlow
A TensorFlow library for applied reinforcement learning
MiniSom is a minimalistic implementation of the Self Organizing Maps
Scripts for fine-tuning Meta Llama3 with composable FSDP & PEFT method
Library to help with training and evaluating neural networks
mlpack: a scalable C++ machine learning library
High-performance library for gradient boosting on decision trees
Detecting silent model failure. NannyML estimates performance
Python package built to ease deep learning on graph
Geometric deep learning extension library for PyTorch
A library for accelerating Transformer models on NVIDIA GPUs
A lightweight 3D Morphable Face Model library in modern C++
Fast Python collaborative filtering for implicit feedback datasets