DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.

Tensors and Dynamic neural networks in Python with strong GPU acceleration

PennyLane is a cross-platform Python library for quantum computing, quantum machine learning, and quantum chemistry. Train a quantum computer the same way as a neural network.

An even simpler way to generate text with LLMs.

🤖 Collect practical AI repos, tools, websites, papers and tutorials on AI. 实用的AI百宝箱 💎

Large Language Model Text Generation Inference

Streamlit — A faster way to build and share data apps.

OpenVINO™ is an open-source toolkit for optimizing and deploying AI inference

Code for "Effect of equivariance on training dynamics"

Paper reading notes on Deep Learning and Machine Learning

A GPU-accelerated library containing highly optimized building blocks and an execution engine for data processing to accelerate deep learning training and inference applications.

Python-based research interface for blackbox and hyperparameter optimization, based on the internal Google Vizier Service.

A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit floating point (FP8) precision on Hopper and Ada GPUs, to provide better performance with lower memory utilization in both training and inference.

Website for the Deep Learning IndabaX South Africa.

Enabling PyTorch on XLA Devices (e.g. Google TPU)

Transparent and reproducible medical image processing pipelines in Python.

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

Neural Network Compression Framework for enhanced OpenVINO™ inference

A flexible, modular, and easy to use library to facilitate federated learning research and development in healthcare settings

moai is a PyTorch-based AI Model Development Kit (MDK) created to improve data-driven model workflows, design and reproducibility.