Python Stream Processing
A modular, primitive-first, python-first PyTorch library
Uniform Manifold Approximation and Projection
A unified, comprehensive and efficient recommendation library
Time series forecasting with PyTorch
A library for accelerating Transformer models on NVIDIA GPUs
Trainable models and NN optimization tools
A library for easily evaluating machine learning models and datasets
AI discovers 520000 stable inorganic crystal structures for research
Training data (data labeling, annotation, workflow) for all data types
Implementation of Imagen, Google's Text-to-Image Neural Network
Functional Machine Learning
Python Optimal Transport
Multi-class confusion matrix library in Python
A Python library for learning and evaluating knowledge graph embedding
Source code of PyGAD, Python 3 library for building genetic algorithms
Topic Modelling for Humans
MiniSom is a minimalistic implementation of the Self Organizing Maps
Data manipulation and transformation for audio signal processing
Fault-tolerant, highly scalable GPU orchestration
Scripts for fine-tuning Meta Llama3 with composable FSDP & PEFT method
Machine learning metrics for distributed, scalable PyTorch application
Optax is a gradient processing and optimization library for JAX
DoWhy is a Python library for causal inference
A simple but complete full-attention transformer