JAX

jax-ml/jax

Composable transforms for NumPy programs

PersonalEnterpriseAI & LLM toolingLibrary / SDKPermissive
docs.jax.dev
Preview of JAX
Preview of JAX at docs.jax.dev

About

Autodiff, vectorization and JIT compilation to GPU and TPU.

From the repository: “Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more”

Browser UseLet AI agents use a web browserLibrary / SDK89
Crawl4AIWeb crawler that turns sites into LLM-ready MarkdownDocker / self-hostLibrary / SDK88
MarkItDownConvert Office files and PDFs to Markdown for LLMsCLILibrary / SDK87
HeadroomCompress tool output before it reaches the LLMLibrary / SDK87

Topics