Top pick
Composable transformations (grad, vmap, pmap) plus NumPy-like API for high-performance ML research on accelerators.
frameworkresearchgputpupython
Browse & filter
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.
1 tool match your filters