Tensors and Dynamic neural networks in Python with strong GPU acceleration

An open platform for training, serving, and evaluating large language models. Release repo for Vicuna and Chatbot Arena.

Data validation using Python type hints

Multilingual Sentence & Image Embeddings with BERT

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

A Heterogeneous Benchmark for Information Retrieval. Easy to use, evaluate your models across 15+ diverse IR datasets.