Python interface to automatically formulate Machine Learning models into Mixed-Integer Programs

A high-throughput and memory-efficient inference and serving engine for LLMs

NEW - YOLOv8 🚀 in PyTorch > ONNX > OpenVINO > CoreML > TFLite

Official implementation of DrugGEN

🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.

A Fundamental End-to-End Speech Recognition Toolkit and Open Source SOTA Pretrained Models, Supporting Speech Recognition, Voice Activity Detection, Text Post-processing etc.

Train, Evaluate, Optimize, Deploy Computer Vision Models via OpenVINO™

Ray is a unified framework for scaling AI and Python applications. Ray consists of a core distributed runtime and a set of AI Libraries for accelerating ML workloads.

Large Language Model Text Generation Inference

📘 The MLOps stack component for experiment tracking

Pretrain, finetune and deploy AI models on multiple GPUs, TPUs with zero code changes.

🤗 Diffusers: State-of-the-art diffusion models for image and audio generation in PyTorch and FLAX.

Fast and Accurate ML in 3 Lines of Code

Aim 💫 — An easy-to-use & supercharged open-source experiment tracker.

🔍 LLM orchestration framework to build customizable, production-ready LLM applications. Connect components (models, vector DBs, file converters) to pipelines or agents that can interact with your data. With advanced retrieval methods, it's best suited for building RAG, question answering, semantic search or conversational agent chatbots.

Code for our paper "Generative Adversarial Network with Soft-Dynamic Time Warping and Parallel Reconstruction for Energy Time Series Anomaly Detection" and its extension.

Lightning support for Intel Habana accelerators.

A Long Term Training On Artificial Intelligence

Model Compression Toolkit (MCT) is an open source project for neural network model optimization under efficient, constrained hardware. This project provides researchers, developers, and engineers advanced quantization and compression tools for deploying state-of-the-art neural networks.

Avalanche: an End-to-End Library for Continual Learning based on PyTorch.