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
Geometric deep learning extension library for PyTorch
AI discovers 520000 stable inorganic crystal structures for research
The most intuitive, flexible, way for researchers to build models
A unified interface for distributed computing
Hub of ready-to-use datasets for ML models
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
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
A simple but complete full-attention transformer
Superfast AI decision making and processing of multi-modal data
Algorithmic Trading in Python with Machine Learning