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.
Solve end to end problems using Llama model family
AI discovers 520000 stable inorganic crystal structures for research
A game theoretic approach to explain the output of ml models
A toolkit to optimize ML models for deployment for Keras & TensorFlow
Time series forecasting with PyTorch
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
Open-source tool designed to enhance the efficiency of workloads
Implementation of Denoising Diffusion Probabilistic Model in Pytorch
Fundamentals of Machine Learning and Deep Learning