Trainable models and NN optimization tools
Decentralized deep learning in PyTorch. Built to train models
Convert TensorFlow, Keras, Tensorflow.js and Tflite models to ONNX
Streamline your ML workflow
Open deep learning compiler stack for cpu, gpu, etc.
Geometric deep learning extension library for PyTorch
Solve end to end problems using Llama model family
All course materials for the Zero to Mastery Machine Learning
AI discovers 520000 stable inorganic crystal structures for research
A toolkit to optimize ML models for deployment for Keras & TensorFlow
Time series forecasting with PyTorch
A game theoretic approach to explain the output of ml models
High-performance library for gradient boosting on decision trees
Prevent PyTorch's `CUDA error: out of memory` in just 1 line of code
Detecting silent model failure. NannyML estimates performance
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
An Easy-to-use, Scalable and High-performance RLHF Framework
MiniSom is a minimalistic implementation of the Self Organizing Maps
Standalone, small, language-neutral
Pretrained (Language) Models for Probabilistic Time Series Forecasting
A high performance implementation of HDBSCAN clustering
Python library for decision tree visualization & model interpretation
SkyPilot: Run AI and batch jobs on any infra