A Python library for learning and evaluating knowledge graph embedding
Fast Python collaborative filtering for implicit feedback datasets
Topic Modelling for Humans
150+ quantitative finance Python programs
Helps scientists define testable, modular, self-documenting dataflow
AI agents autonomously run and improve ML experiments overnight
A Python library for audio
State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX
Detecting silent model failure. NannyML estimates performance
CLI wrapper around aws-encryption-sdk-python
Multi-Joint dynamics with Contact. A general purpose physics simulator
Some ipython notebooks implementing AI algorithms
An Easy-to-use, Scalable and High-performance RLHF Framework
MiniSom is a minimalistic implementation of the Self Organizing Maps
Pretrained (Language) Models for Probabilistic Time Series Forecasting
Explainability and Interpretability to Develop Reliable ML models
A high performance implementation of HDBSCAN clustering
Python framework for adversarial attacks, and data augmentation
Hummingbird compiles trained ML models into tensor computation
SkyPilot: Run AI and batch jobs on any infra
We write your reusable computer vision tools
Synthetic Data Generation for tabular, relational and time series data
Implementation of Denoising Diffusion Probabilistic Model in Pytorch
Graph Neural Network Library for PyTorch
Open source annotation tool for machine learning practitioners