State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX
A simple but complete full-attention transformer
Feature engineering package with sklearn like functionality
Quickly get started with AI theory and practical applications
A PyTorch-based Speech Toolkit
Statistical library designed to fill the void in Python's time series
Flower: A Friendly Federated Learning Framework
An agentic Machine Learning Engineer
Models for the spaCy Natural Language Processing (NLP) library
A game theoretic approach to explain the output of ml models
Deep learning optimization library making distributed training easy
A very simple framework for state-of-the-art NLP
A library for accelerating Transformer models on NVIDIA GPUs
Scalable and user friendly neural forecasting algorithms.
An MLOps framework to package, deploy, monitor and manage models
ktrain is a Python library that makes deep learning AI more accessible
BitNet: Scaling 1-bit Transformers for Large Language Models
Implementation of BEVFormer, a camera-only framework
Unofficial Pytorch implementation / replication of Alphafold2
Deep learning PyTorch library for time series forecasting
An Open-Source Framework for Prompt-Learning
Code Repository for Machine Learning with PyTorch and Scikit-Learn
CPU/GPU inference server for Hugging Face transformer models
Fast & easy transfer learning for NLP