A framework for few-shot evaluation of language models.

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

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.

Elegant easy-to-use neural networks in JAX. https://docs.kidger.site/equinox/

Python & command-line tool to gather text on the Web: web crawling/scraping, extraction of text, metadata, comments

Convert HTML to Markdown-formatted text.

A machine learning compiler for GPUs, CPUs, and ML accelerators

Experiments with Jax and TPUs for Foundation Models - modified tokens for lakh

Simple, Elegant, Typed Argument Parsing with argparse

Visualize, create, and operate on JAX PyTree in the most intuitive way possible.

Code for the paper "Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformer"

Task-based datasets, preprocessing, and evaluation for sequence models.

Collection of Singularity Definition Files and .sif Images for Reproducibility.

PyTorch extensions for high performance and large scale training.

A specification that python filesystems should adhere to.

hyperparameter management

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

A library to create and manage configuration files, especially for machine learning projects.

Rust bindings to the MLIR C API.

The Scala 3 compiler, also known as Dotty.

Representation and Reference Lowering of ONNX Models in MLIR Compiler Infrastructure

Microsoft Azure Storage Client Library for C++