Open-source tools for prompt testing and experimentation, with support for both LLMs (e.g. OpenAI, LLaMA) and vector databases (e.g. Chroma, Weaviate, LanceDB).

Benchmarking and profiling various video reading options for python

Dispatch and distribute your ML training to "serverless" clusters in Python, like PyTorch for ML infra. Iterable, debuggable, multi-cloud/on-prem, identical across research and production.

TorchBench is a collection of open source benchmarks used to evaluate PyTorch performance.

Datasets, Transforms and Models specific to Computer Vision

Models, data loaders and abstractions for language processing, powered by PyTorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration