Your open-source LLM evaluation toolkit
Trainable models and NN optimization tools
Medical imaging toolkit for deep learning
A comprehensive set of fairness metrics for datasets
The easiest way to use deep metric learning in your application
AI discovers 520000 stable inorganic crystal structures for research
The most intuitive, flexible, way for researchers to build models
Create videos with Stable Diffusion
Hub of ready-to-use datasets for ML models
Lazy Predict help build a lot of basic models without much code
The machine learning toolkit for time series analysis in Python
A unified interface for distributed computing
Source code of PyGAD, Python 3 library for building genetic algorithms
Elyra extends JupyterLab with an AI centric approach
TimeGPT-1: production ready pre-trained Time Series Foundation Model
A high performance implementation of HDBSCAN clustering
Pytorch domain library for recommendation systems
Simulation of spiking neural networks (SNNs) using PyTorch
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
Uplift modeling and causal inference with machine learning algorithms
An easy-to-use & supercharged open-source experiment tracker
DoWhy is a Python library for causal inference