Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)

Example Python package with a C++-based C-extension

AI orchestration framework to build customizable, production-ready LLM applications. Connect components (models, vector DBs, file converters) to pipelines or agents that can interact with your data…

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

ALFRED - A Benchmark for Interpreting Grounded Instructions for Everyday Tasks

Hydra is a framework for elegantly configuring complex applications

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)