An open-source, code-first Typescript toolkit for building, evaluating, and deploying sophisticated AI agents with flexibility and control.

A library for generative social simulation

TimesFM (Time Series Foundation Model) is a pretrained time-series foundation model developed by Google Research for time-series forecasting.

Pax is a Jax-based machine learning framework for training large scale models. Pax allows for advanced and fully configurable experimentation and parallelization, and has demonstrated industry leading model flop utilization rates.

Code for the paper "Defeating Prompt Injections by Design"

A JAX research toolkit for building, editing, and visualizing neural networks.

Minimal and scalable research codebase in JAX, designed for rapid iteration on frontier research in LLM and other autoregressive models.

Machine learning for machine code.

A high-throughput and memory-efficient inference and serving engine for LLMs