仓库

jax-ml 的仓库

5 个支持的仓库

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

最近提交 2026年8月16日

 (36,165 stars) (3,737 forks) (4 个已索引 issue) (4 个开放 good first issue)

最近提交 2026年8月3日

 (310 stars) (48 forks) (0 个已索引 issue) (0 个开放 good first issue)

最近提交 2026年1月9日

 (6 stars) (2 forks) (0 个已索引 issue) (0 个开放 good first issue)

A stand-alone implementation of several NumPy dtype extensions used in machine learning.

最近提交 2026年8月13日

 (357 stars) (59 forks) (0 个已索引 issue) (0 个开放 good first issue)

Oryx is a library for probabilistic programming and deep learning built on top of Jax.

最近提交 2026年7月8日

 (320 stars) (13 forks) (1 个已索引 issue) (1 个开放 good first issue)