TNCO is a heuristic tool that optimizes tensor network contraction paths.

A modular framework for evaluating weather forecasts

Tooling for agile modeling on large machine perception embedding databases.

Dataclasses manipulated as numpy arrays (with batching, reshape, slicing,...)

Python library which enables complex compositions of language models such as scratchpads, chain of thought, tool use, selection-inference, and more.

A JAX library for building lattice-based speech transducer models

A benchmark for the next generation of data-driven global weather models.

The Zebrafish Activity Prediction Benchmark measures progress on the problem of predicting cellular-resolution neural activity throughout an entire vertebrate brain.