The LLM Evaluation Framework

A Library for Advanced Deep Time Series Models for General Time Series Analysis.

Command line driven CI frontend and development task automation tool.

Utilities for interacting with PyPI

Case study submission for the DataCamp Associate Data Analyst Certificate Exam.

Amazon SageMaker Local Mode Examples

Python Solutions of LeetCode Problems. Daily Update. If you like this, please leave me a star. ★

OCR, layout analysis, reading order, table recognition in 90+ languages

Easily use and train state of the art late-interaction retrieval methods (ColBERT) in any RAG pipeline. Designed for modularity and ease-of-use, backed by research.

500 AI Machine learning Deep learning Computer vision NLP Projects with code

A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit and 4-bit floating point (FP8 and FP4) precision on Hopper, Ada and Blackwell GPUs, to provide better performance…

An implementation of model parallel autoregressive transformers on GPUs, based on the Megatron and DeepSpeed libraries

A simple, performant and scalable Jax LLM!

Mesh TensorFlow: Model Parallelism Made Easier

Making large AI models cheaper, faster and more accessible

Ongoing research training transformer models at scale

A scalable generative AI framework built for researchers and developers working on Large Language Models, Multimodal, and Speech AI (Automatic Speech Recognition and Text-to-Speech)

A PyTorch native platform for training generative AI models

PyTorch native post-training library

Ongoing research training transformer language models at scale, including: BERT & GPT-2

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

A high-performance topological machine learning toolbox in Python

Feature engineering package with sklearn like functionality

A library of sklearn compatible categorical variable encoders

A high performance implementation of HDBSCAN clustering.

Topic modeling with latent Dirichlet allocation using Gibbs sampling

Genetic Programming in Python, with a scikit-learn inspired API

Concrete ML: Privacy Preserving ML framework using Fully Homomorphic Encryption (FHE), built on top of Concrete, with bindings to traditional ML frameworks.