Top pick
Composable transformations (grad, vmap, pmap) plus NumPy-like API for high-performance ML research on accelerators.
frameworkresearchgputpupython
Filter by platform, license text, maturity, maintenance cadence, and editorial tags like privacy-focused or self-hosted. Search matches names, summaries, tags, and use cases.
4 tools match your filters
Composable transformations (grad, vmap, pmap) plus NumPy-like API for high-performance ML research on accelerators.
High-level multi-backend deep learning API (TensorFlow, JAX, PyTorch) focused on ergonomics and fast iteration.
Progressive JavaScript framework for UIs.
Zsh configuration framework emphasizing speed and organization.