https://github.com/google/jax

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
https://github.com/google/jax

jax

Last synced: about 2 months ago
JSON representation

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

• Host: GitHub
• URL: https://github.com/google/jax
• Owner: google
• License: apache-2.0
• Created: 2018-10-25T21:25:02.000Z (almost 6 years ago)
• Default Branch: main
• Last Pushed: 2024-05-01T14:11:15.000Z (5 months ago)
• Last Synced: 2024-05-01T14:37:41.005Z (5 months ago)
• Topics: jax
• Language: Python
• Homepage: http://jax.readthedocs.io/
• Size: 79.1 MB
• Stars: 28,003
• Watchers: 322
• Forks: 2,565
• Open Issues: 1,675
• Metadata Files:
  • Readme: README.md
  • Changelog: CHANGELOG.md
  • Contributing: CONTRIBUTING.md
  • License: LICENSE
  • Citation: CITATION.bib
  • Authors: AUTHORS

Awesome Lists containing this project

• awesomeLibrary - jax - Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more (语言资源库 / python)
• awesome-colab-project - JAX
• awesome-real-time-AI - JAX
• awesome-python-machine-learning-resources - GitHub - 24% open · ⏱️ 26.08.2022): (机器学习框架)
• awesome-llmops - Jax - performance machine learning research. |  | (Training / Frameworks for Training)
• awesome-mojo - Google JAX
• awesome-rainmana - google/jax - Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more (Python)
• awesome-llm-eval - Jax - Autograd and XLA for high-performance machine learning research. (Frameworks-for-Training / Popular-LLM)
• awesome-starts - google/jax - Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more (Python)
• AiTreasureBox - google/jax - 09-13_29801_4](https://img.shields.io/github/stars/google/jax.svg)|Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more| (Repos)
• stars - google/jax - Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more (Python)
• awesome-list - JAX - Automatical differentiation for native Python and NumPy functions, with GPU support. (Machine Learning Framework / General Purpose Framework)
• awesome-ML-NLP - JAX - Autograd and XLA, brought together for high-performance machine learning research (Libraries, Softwares)
• awesome-econ - JAX
• awesome-google-colab - JAX - Composable transformations of Python+NumPy programs (Technologies)
• awesome-starred - google/jax - Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more (others)
• my-awesome - google/jax - 09 star:29.7k fork:2.7k Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more (Python)
• awesome-systematic-trading - JAX - commit/google/jax/main)  - Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more (Basic Components / Fundamental libraries)
• awesome-production-machine-learning - Jax - Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more. (Optimized Computation)
• awesome-sciml - google/jax: Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more