JAX
约 1 分钟阅读
- google/jaxGitHubgoogle/jaxgoogle/jax · Apache-2.0, Python, C++ · Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more笔记：JAX
- Apache-2.0, Python, C++
- Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
- 参考
- vopani/jaxtonGitHubvopani/jaxtongoogle/jax · Apache-2.0, Python, C++ · Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more笔记：JAX
- exercises to learn JAX
- felafax/felafaxGitHubfelafax/felafax
- PyTorch is dead. Long live JAX.Linkneel04.github.io/my-website/blog/pytorch_rantgoogle/jax · Apache-2.0, Python, C++ · Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more笔记：JAX
- vopani/jaxtonGitHubvopani/jaxtongoogle/jax · Apache-2.0, Python, C++ · Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more笔记：JAX
关联信息
反向链接和本文引用的外部资料。
反向链接
- ML Awesome笔记 · JAX
LLM · LLM Awesome · Diffusion · Diffusion Awesome · NLP · NLP Awesome · GAN · img2img · Framework · PyTorch · by Meta · tinygrad/tinygrad
References
GitHub
3 条- felafax/felafaxgithub.com/felafax/felafax
- google/jaxgithub.com/google/jax
- vopani/jaxtongithub.com/vopani/jaxton
其他外链
1 条- neel04.github.io/my-website/blog/pytorch_rantneel04.github.io/my-website/blog/pytorch_rant