René's URL Explorer Experiment

Title: GitHub - sjtechdev/jax: Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

Open Graph Title: GitHub - sjtechdev/jax: Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

X Title: GitHub - sjtechdev/jax: Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

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

Open Graph Description: Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more - sjtechdev/jax

X Description: Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more - sjtechdev/jax

Opengraph URL: https://github.com/sjtechdev/jax

X: @github

direct link

Domain: patch-diff.githubusercontent.com

route-pattern	/:user_id/:repository
route-controller	files
route-action	disambiguate
fetch-nonce	v2:0da4f215-448f-4001-7182-2632bc110684
current-catalog-service-hash	f3abb0cc802f3d7b95fc8762b94bdcb13bf39634c40c357301c4aa1d67a256fb
request-id	86C6:121EA9:6C8B6:9067A:6981B804
html-safe-nonce	f725a06aac2d8d97f057dc3324b700e8db71ee36ceb0b0d5832023b45ee1f692
visitor-payload	eyJyZWZlcnJlciI6IiIsInJlcXVlc3RfaWQiOiI4NkM2OjEyMUVBOTo2QzhCNjo5MDY3QTo2OTgxQjgwNCIsInZpc2l0b3JfaWQiOiI4MzEyMzkzMTMyNzkyNzIzNDYwIiwicmVnaW9uX2VkZ2UiOiJpYWQiLCJyZWdpb25fcmVuZGVyIjoiaWFkIn0=
visitor-hmac	c267b411b64fe6fe8c8c45593e286d2a8f0616bcee9337d1b34c3ba042137228
hovercard-subject-tag	repository:598418480
github-keyboard-shortcuts	repository,copilot
google-site-verification	Apib7-x98H0j5cPqHWwSMm6dNU4GmODRoqxLiDzdx9I
octolytics-url	https://collector.github.com/github/collect
analytics-location	//
fb:app_id	1401488693436528
apple-itunes-app	app-id=1477376905, app-argument=https://github.com/sjtechdev/jax
twitter:image	https://opengraph.githubassets.com/e2471882f9fd0610cd6185cbccc4be9d8362f1335e960654c8bd771f889b6a81/sjtechdev/jax
twitter:card	summary_large_image
og:image	https://opengraph.githubassets.com/e2471882f9fd0610cd6185cbccc4be9d8362f1335e960654c8bd771f889b6a81/sjtechdev/jax
og:image:alt	Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more - sjtechdev/jax
og:image:width	1200
og:image:height	600
og:site_name	GitHub
og:type	object
hostname	github.com
expected-hostname	github.com
None	e137814e266030874fd2c86863529d0622b13889eeda04148c57654b6ea84ad6
turbo-cache-control	no-preview
go-import	github.com/sjtechdev/jax git https://github.com/sjtechdev/jax.git
octolytics-dimension-user_id	20032028
octolytics-dimension-user_login	sjtechdev
octolytics-dimension-repository_id	598418480
octolytics-dimension-repository_nwo	sjtechdev/jax
octolytics-dimension-repository_public	true
octolytics-dimension-repository_is_fork	true
octolytics-dimension-repository_parent_id	154739597
octolytics-dimension-repository_parent_nwo	jax-ml/jax
octolytics-dimension-repository_network_root_id	154739597
octolytics-dimension-repository_network_root_nwo	jax-ml/jax
turbo-body-classes	logged-out env-production page-responsive
disable-turbo	false
browser-stats-url	https://api.github.com/_private/browser/stats
browser-errors-url	https://api.github.com/_private/browser/errors
release	dd58d68a7813bbec9c91422c8c35f4af33832d70
ui-target	full
theme-color	#1e2327
color-scheme	light dark

Links:

Skip to content	https://patch-diff.githubusercontent.com/sjtechdev/jax#start-of-content
	https://patch-diff.githubusercontent.com/
Sign in	https://patch-diff.githubusercontent.com/login?return_to=https%3A%2F%2Fgithub.com%2Fsjtechdev%2Fjax
GitHub CopilotWrite better code with AI	https://github.com/features/copilot
GitHub SparkBuild and deploy intelligent apps	https://github.com/features/spark
GitHub ModelsManage and compare prompts	https://github.com/features/models
MCP RegistryNewIntegrate external tools	https://github.com/mcp
ActionsAutomate any workflow	https://github.com/features/actions
CodespacesInstant dev environments	https://github.com/features/codespaces
IssuesPlan and track work	https://github.com/features/issues
Code ReviewManage code changes	https://github.com/features/code-review
GitHub Advanced SecurityFind and fix vulnerabilities	https://github.com/security/advanced-security
Code securitySecure your code as you build	https://github.com/security/advanced-security/code-security
Secret protectionStop leaks before they start	https://github.com/security/advanced-security/secret-protection
Why GitHub	https://github.com/why-github
Documentation	https://docs.github.com
Blog	https://github.blog
Changelog	https://github.blog/changelog
Marketplace	https://github.com/marketplace
View all features	https://github.com/features
Enterprises	https://github.com/enterprise
Small and medium teams	https://github.com/team
Startups	https://github.com/enterprise/startups
Nonprofits	https://github.com/solutions/industry/nonprofits
App Modernization	https://github.com/solutions/use-case/app-modernization
DevSecOps	https://github.com/solutions/use-case/devsecops
DevOps	https://github.com/solutions/use-case/devops
CI/CD	https://github.com/solutions/use-case/ci-cd
View all use cases	https://github.com/solutions/use-case
Healthcare	https://github.com/solutions/industry/healthcare
Financial services	https://github.com/solutions/industry/financial-services
Manufacturing	https://github.com/solutions/industry/manufacturing
Government	https://github.com/solutions/industry/government
View all industries	https://github.com/solutions/industry
View all solutions	https://github.com/solutions
AI	https://github.com/resources/articles?topic=ai
Software Development	https://github.com/resources/articles?topic=software-development
DevOps	https://github.com/resources/articles?topic=devops
Security	https://github.com/resources/articles?topic=security
View all topics	https://github.com/resources/articles
Customer stories	https://github.com/customer-stories
Events & webinars	https://github.com/resources/events
Ebooks & reports	https://github.com/resources/whitepapers
Business insights	https://github.com/solutions/executive-insights
GitHub Skills	https://skills.github.com
Documentation	https://docs.github.com
Customer support	https://support.github.com
Community forum	https://github.com/orgs/community/discussions
Trust center	https://github.com/trust-center
Partners	https://github.com/partners
GitHub SponsorsFund open source developers	https://github.com/sponsors
Security Lab	https://securitylab.github.com
Maintainer Community	https://maintainers.github.com
Accelerator	https://github.com/accelerator
Archive Program	https://archiveprogram.github.com
Topics	https://github.com/topics
Trending	https://github.com/trending
Collections	https://github.com/collections
Enterprise platformAI-powered developer platform	https://github.com/enterprise
GitHub Advanced SecurityEnterprise-grade security features	https://github.com/security/advanced-security
Copilot for BusinessEnterprise-grade AI features	https://github.com/features/copilot/copilot-business
Premium SupportEnterprise-grade 24/7 support	https://github.com/premium-support
Pricing	https://github.com/pricing
Search syntax tips	https://docs.github.com/search-github/github-code-search/understanding-github-code-search-syntax
documentation	https://docs.github.com/search-github/github-code-search/understanding-github-code-search-syntax
Sign in	https://patch-diff.githubusercontent.com/login?return_to=https%3A%2F%2Fgithub.com%2Fsjtechdev%2Fjax
Sign up	https://patch-diff.githubusercontent.com/signup?ref_cta=Sign+up&ref_loc=header+logged+out&ref_page=%2F%3Cuser-name%3E%2F%3Crepo-name%3E&source=header-repo&source_repo=sjtechdev%2Fjax
Reload	https://patch-diff.githubusercontent.com/sjtechdev/jax
Reload	https://patch-diff.githubusercontent.com/sjtechdev/jax
Reload	https://patch-diff.githubusercontent.com/sjtechdev/jax
sjtechdev	https://patch-diff.githubusercontent.com/sjtechdev
jax	https://patch-diff.githubusercontent.com/sjtechdev/jax
jax-ml/jax	https://patch-diff.githubusercontent.com/jax-ml/jax
Notifications	https://patch-diff.githubusercontent.com/login?return_to=%2Fsjtechdev%2Fjax
Fork 0	https://patch-diff.githubusercontent.com/login?return_to=%2Fsjtechdev%2Fjax
Star 0	https://patch-diff.githubusercontent.com/login?return_to=%2Fsjtechdev%2Fjax
jax.readthedocs.io/	http://jax.readthedocs.io/
Apache-2.0 license	https://patch-diff.githubusercontent.com/sjtechdev/jax/blob/main/LICENSE
0 stars	https://patch-diff.githubusercontent.com/sjtechdev/jax/stargazers
3.4k forks	https://patch-diff.githubusercontent.com/sjtechdev/jax/forks
Branches	https://patch-diff.githubusercontent.com/sjtechdev/jax/branches
Tags	https://patch-diff.githubusercontent.com/sjtechdev/jax/tags
Activity	https://patch-diff.githubusercontent.com/sjtechdev/jax/activity
Star	https://patch-diff.githubusercontent.com/login?return_to=%2Fsjtechdev%2Fjax
Notifications	https://patch-diff.githubusercontent.com/login?return_to=%2Fsjtechdev%2Fjax
Code	https://patch-diff.githubusercontent.com/sjtechdev/jax
Pull requests 0	https://patch-diff.githubusercontent.com/sjtechdev/jax/pulls
Actions	https://patch-diff.githubusercontent.com/sjtechdev/jax/actions
Projects 0	https://patch-diff.githubusercontent.com/sjtechdev/jax/projects
Security 0	https://patch-diff.githubusercontent.com/sjtechdev/jax/security
Insights	https://patch-diff.githubusercontent.com/sjtechdev/jax/pulse
Code	https://patch-diff.githubusercontent.com/sjtechdev/jax
Pull requests	https://patch-diff.githubusercontent.com/sjtechdev/jax/pulls
Actions	https://patch-diff.githubusercontent.com/sjtechdev/jax/actions
Projects	https://patch-diff.githubusercontent.com/sjtechdev/jax/projects
Security	https://patch-diff.githubusercontent.com/sjtechdev/jax/security
Insights	https://patch-diff.githubusercontent.com/sjtechdev/jax/pulse
Branches	https://patch-diff.githubusercontent.com/sjtechdev/jax/branches
Tags	https://patch-diff.githubusercontent.com/sjtechdev/jax/tags
	https://patch-diff.githubusercontent.com/sjtechdev/jax/branches
	https://patch-diff.githubusercontent.com/sjtechdev/jax/tags
14,677 Commits	https://patch-diff.githubusercontent.com/sjtechdev/jax/commits/main/
	https://patch-diff.githubusercontent.com/sjtechdev/jax/commits/main/
.github	https://patch-diff.githubusercontent.com/sjtechdev/jax/tree/main/.github
.github	https://patch-diff.githubusercontent.com/sjtechdev/jax/tree/main/.github
benchmarks	https://patch-diff.githubusercontent.com/sjtechdev/jax/tree/main/benchmarks
benchmarks	https://patch-diff.githubusercontent.com/sjtechdev/jax/tree/main/benchmarks
build	https://patch-diff.githubusercontent.com/sjtechdev/jax/tree/main/build
build	https://patch-diff.githubusercontent.com/sjtechdev/jax/tree/main/build
cloud_tpu_colabs	https://patch-diff.githubusercontent.com/sjtechdev/jax/tree/main/cloud_tpu_colabs
cloud_tpu_colabs	https://patch-diff.githubusercontent.com/sjtechdev/jax/tree/main/cloud_tpu_colabs
docs	https://patch-diff.githubusercontent.com/sjtechdev/jax/tree/main/docs
docs	https://patch-diff.githubusercontent.com/sjtechdev/jax/tree/main/docs
examples	https://patch-diff.githubusercontent.com/sjtechdev/jax/tree/main/examples
examples	https://patch-diff.githubusercontent.com/sjtechdev/jax/tree/main/examples
images	https://patch-diff.githubusercontent.com/sjtechdev/jax/tree/main/images
images	https://patch-diff.githubusercontent.com/sjtechdev/jax/tree/main/images
jax	https://patch-diff.githubusercontent.com/sjtechdev/jax/tree/main/jax
jax	https://patch-diff.githubusercontent.com/sjtechdev/jax/tree/main/jax
jaxlib	https://patch-diff.githubusercontent.com/sjtechdev/jax/tree/main/jaxlib
jaxlib	https://patch-diff.githubusercontent.com/sjtechdev/jax/tree/main/jaxlib
tests	https://patch-diff.githubusercontent.com/sjtechdev/jax/tree/main/tests
tests	https://patch-diff.githubusercontent.com/sjtechdev/jax/tree/main/tests
third_party/ducc	https://patch-diff.githubusercontent.com/sjtechdev/jax/tree/main/third_party/ducc
third_party/ducc	https://patch-diff.githubusercontent.com/sjtechdev/jax/tree/main/third_party/ducc
.bazelrc	https://patch-diff.githubusercontent.com/sjtechdev/jax/blob/main/.bazelrc
.bazelrc	https://patch-diff.githubusercontent.com/sjtechdev/jax/blob/main/.bazelrc
.bazelversion	https://patch-diff.githubusercontent.com/sjtechdev/jax/blob/main/.bazelversion
.bazelversion	https://patch-diff.githubusercontent.com/sjtechdev/jax/blob/main/.bazelversion
.gitignore	https://patch-diff.githubusercontent.com/sjtechdev/jax/blob/main/.gitignore
.gitignore	https://patch-diff.githubusercontent.com/sjtechdev/jax/blob/main/.gitignore
.pre-commit-config.yaml	https://patch-diff.githubusercontent.com/sjtechdev/jax/blob/main/.pre-commit-config.yaml
.pre-commit-config.yaml	https://patch-diff.githubusercontent.com/sjtechdev/jax/blob/main/.pre-commit-config.yaml
.readthedocs.yml	https://patch-diff.githubusercontent.com/sjtechdev/jax/blob/main/.readthedocs.yml
.readthedocs.yml	https://patch-diff.githubusercontent.com/sjtechdev/jax/blob/main/.readthedocs.yml
AUTHORS	https://patch-diff.githubusercontent.com/sjtechdev/jax/blob/main/AUTHORS
AUTHORS	https://patch-diff.githubusercontent.com/sjtechdev/jax/blob/main/AUTHORS
CHANGELOG.md	https://patch-diff.githubusercontent.com/sjtechdev/jax/blob/main/CHANGELOG.md
CHANGELOG.md	https://patch-diff.githubusercontent.com/sjtechdev/jax/blob/main/CHANGELOG.md
CITATION.bib	https://patch-diff.githubusercontent.com/sjtechdev/jax/blob/main/CITATION.bib
CITATION.bib	https://patch-diff.githubusercontent.com/sjtechdev/jax/blob/main/CITATION.bib
CONTRIBUTING.md	https://patch-diff.githubusercontent.com/sjtechdev/jax/blob/main/CONTRIBUTING.md
CONTRIBUTING.md	https://patch-diff.githubusercontent.com/sjtechdev/jax/blob/main/CONTRIBUTING.md
LICENSE	https://patch-diff.githubusercontent.com/sjtechdev/jax/blob/main/LICENSE
LICENSE	https://patch-diff.githubusercontent.com/sjtechdev/jax/blob/main/LICENSE
README.md	https://patch-diff.githubusercontent.com/sjtechdev/jax/blob/main/README.md
README.md	https://patch-diff.githubusercontent.com/sjtechdev/jax/blob/main/README.md
WORKSPACE	https://patch-diff.githubusercontent.com/sjtechdev/jax/blob/main/WORKSPACE
WORKSPACE	https://patch-diff.githubusercontent.com/sjtechdev/jax/blob/main/WORKSPACE
conftest.py	https://patch-diff.githubusercontent.com/sjtechdev/jax/blob/main/conftest.py
conftest.py	https://patch-diff.githubusercontent.com/sjtechdev/jax/blob/main/conftest.py
mypy.ini	https://patch-diff.githubusercontent.com/sjtechdev/jax/blob/main/mypy.ini
mypy.ini	https://patch-diff.githubusercontent.com/sjtechdev/jax/blob/main/mypy.ini
pylintrc	https://patch-diff.githubusercontent.com/sjtechdev/jax/blob/main/pylintrc
pylintrc	https://patch-diff.githubusercontent.com/sjtechdev/jax/blob/main/pylintrc
pytest.ini	https://patch-diff.githubusercontent.com/sjtechdev/jax/blob/main/pytest.ini
pytest.ini	https://patch-diff.githubusercontent.com/sjtechdev/jax/blob/main/pytest.ini
setup.cfg	https://patch-diff.githubusercontent.com/sjtechdev/jax/blob/main/setup.cfg
setup.cfg	https://patch-diff.githubusercontent.com/sjtechdev/jax/blob/main/setup.cfg
setup.py	https://patch-diff.githubusercontent.com/sjtechdev/jax/blob/main/setup.py
setup.py	https://patch-diff.githubusercontent.com/sjtechdev/jax/blob/main/setup.py
README	https://patch-diff.githubusercontent.com/sjtechdev/jax
Contributing	https://patch-diff.githubusercontent.com/sjtechdev/jax
License	https://patch-diff.githubusercontent.com/sjtechdev/jax
	https://raw.githubusercontent.com/google/jax/main/images/jax_logo_250px.png
	https://patch-diff.githubusercontent.com/sjtechdev/jax#jax-autograd-and-xla
	https://github.com/google/jax/actions/workflows/ci-build.yaml/badge.svg
	https://camo.githubusercontent.com/4f6daec073f48ea3b68546a867ef432dd02601175ce2a012273686cae02091b9/68747470733a2f2f696d672e736869656c64732e696f2f707970692f762f6a6178
Quickstart	https://patch-diff.githubusercontent.com/sjtechdev/jax#quickstart-colab-in-the-cloud
Transformations	https://patch-diff.githubusercontent.com/sjtechdev/jax#transformations
Install guide	https://patch-diff.githubusercontent.com/sjtechdev/jax#installation
Neural net libraries	https://patch-diff.githubusercontent.com/sjtechdev/jax#neural-network-libraries
Change logs	https://jax.readthedocs.io/en/latest/changelog.html
Reference docs	https://jax.readthedocs.io/en/latest/
	https://patch-diff.githubusercontent.com/sjtechdev/jax#what-is-jax
Autograd	https://github.com/hips/autograd
XLA	https://www.tensorflow.org/xla
Autograd	https://github.com/hips/autograd
grad	https://patch-diff.githubusercontent.com/sjtechdev/jax#automatic-differentiation-with-grad
XLA	https://www.tensorflow.org/xla
jit	https://patch-diff.githubusercontent.com/sjtechdev/jax#compilation-with-jit
pmap	https://patch-diff.githubusercontent.com/sjtechdev/jax#spmd-programming-with-pmap
composable function transformations	https://patch-diff.githubusercontent.com/sjtechdev/jax#transformations
grad	https://patch-diff.githubusercontent.com/sjtechdev/jax#automatic-differentiation-with-grad
jit	https://patch-diff.githubusercontent.com/sjtechdev/jax#compilation-with-jit
vmap	https://patch-diff.githubusercontent.com/sjtechdev/jax#auto-vectorization-with-vmap
pmap	https://patch-diff.githubusercontent.com/sjtechdev/jax#spmd-programming-with-pmap
sharp edges	https://jax.readthedocs.io/en/latest/notebooks/Common_Gotchas_in_JAX.html
reporting bugs	https://github.com/google/jax/issues
	https://patch-diff.githubusercontent.com/sjtechdev/jax#contents
Quickstart: Colab in the Cloud	https://patch-diff.githubusercontent.com/sjtechdev/jax#quickstart-colab-in-the-cloud
Transformations	https://patch-diff.githubusercontent.com/sjtechdev/jax#transformations
Current gotchas	https://patch-diff.githubusercontent.com/sjtechdev/jax#current-gotchas
Installation	https://patch-diff.githubusercontent.com/sjtechdev/jax#installation
Neural net libraries	https://patch-diff.githubusercontent.com/sjtechdev/jax#neural-network-libraries
Citing JAX	https://patch-diff.githubusercontent.com/sjtechdev/jax#citing-jax
Reference documentation	https://patch-diff.githubusercontent.com/sjtechdev/jax#reference-documentation
	https://patch-diff.githubusercontent.com/sjtechdev/jax#quickstart-colab-in-the-cloud
The basics: NumPy on accelerators, grad for differentiation, jit for compilation, and vmap for vectorization	https://jax.readthedocs.io/en/latest/notebooks/quickstart.html
Training a Simple Neural Network, with TensorFlow Dataset Data Loading	https://colab.research.google.com/github/google/jax/blob/main/docs/notebooks/neural_network_with_tfds_data.ipynb
Cloud TPU Colabs	https://github.com/google/jax/tree/main/cloud_tpu_colabs
The Autodiff Cookbook, Part 1: easy and powerful automatic differentiation in JAX	https://jax.readthedocs.io/en/latest/notebooks/autodiff_cookbook.html
Common gotchas and sharp edges	https://jax.readthedocs.io/en/latest/notebooks/Common_Gotchas_in_JAX.html
full list of notebooks	https://github.com/google/jax/tree/main/docs/notebooks
the mini-libraries in jax.example_libraries	https://github.com/google/jax/tree/main/jax/example_libraries/README.md
stax for building neural networks	https://github.com/google/jax/tree/main/jax/example_libraries/README.md#neural-net-building-with-stax
optimizers for first-order stochastic optimization	https://github.com/google/jax/tree/main/jax/example_libraries/README.md#first-order-optimization
examples	https://github.com/google/jax/tree/main/examples
	https://patch-diff.githubusercontent.com/sjtechdev/jax#transformations
	https://patch-diff.githubusercontent.com/sjtechdev/jax#automatic-differentiation-with-grad
Autograd	https://github.com/hips/autograd
grad	https://jax.readthedocs.io/en/latest/jax.html#jax.grad
jax.vjp	https://jax.readthedocs.io/en/latest/jax.html#jax.vjp
jax.jvp	https://jax.readthedocs.io/en/latest/jax.html#jax.jvp
full Hessian matrices	https://jax.readthedocs.io/en/latest/jax.html#jax.hessian
Autograd	https://github.com/hips/autograd
reference docs on automatic differentiation	https://jax.readthedocs.io/en/latest/jax.html#automatic-differentiation
JAX Autodiff Cookbook	https://jax.readthedocs.io/en/latest/notebooks/autodiff_cookbook.html
	https://patch-diff.githubusercontent.com/sjtechdev/jax#compilation-with-jit
jit	https://jax.readthedocs.io/en/latest/jax.html#just-in-time-compilation-jit
Gotchas Notebook	https://jax.readthedocs.io/en/latest/notebooks/Common_Gotchas_in_JAX.html#python-control-flow-+-JIT
	https://patch-diff.githubusercontent.com/sjtechdev/jax#auto-vectorization-with-vmap
vmap	https://jax.readthedocs.io/en/latest/jax.html#vectorization-vmap
	https://patch-diff.githubusercontent.com/sjtechdev/jax#spmd-programming-with-pmap
pmap	https://jax.readthedocs.io/en/latest/jax.html#parallelization-pmap
collective communication operations	https://jax.readthedocs.io/en/latest/jax.lax.html#parallel-operators
nest pmap functions	https://colab.research.google.com/github/google/jax/blob/main/cloud_tpu_colabs/Pmap_Cookbook.ipynb#scrollTo=MdRscR5MONuN
SPMD Cookbook	https://colab.research.google.com/github/google/jax/blob/main/cloud_tpu_colabs/Pmap_Cookbook.ipynb
SPMD MNIST classifier from scratch example	https://github.com/google/jax/blob/main/examples/spmd_mnist_classifier_fromscratch.py
	https://patch-diff.githubusercontent.com/sjtechdev/jax#current-gotchas
Gotchas Notebook	https://jax.readthedocs.io/en/latest/notebooks/Common_Gotchas_in_JAX.html
pure functions	https://en.wikipedia.org/wiki/Pure_function
referential transparency	https://en.wikipedia.org/wiki/Referential_transparency
In-place mutating updates of arrays	https://jax.readthedocs.io/en/latest/notebooks/Common_Gotchas_in_JAX.html#in-place-updates
there are functional alternatives	https://jax.readthedocs.io/en/latest/jax.ops.html
Random numbers are different	https://jax.readthedocs.io/en/latest/notebooks/Common_Gotchas_in_JAX.html#random-numbers
good reasons	https://github.com/google/jax/blob/main/docs/jep/263-prng.md
convolution operators	https://jax.readthedocs.io/en/latest/notebooks/convolutions.html
to enable double-precision	https://jax.readthedocs.io/en/latest/notebooks/Common_Gotchas_in_JAX.html#double-64bit-precision
constrain how you can use Python control flow	https://jax.readthedocs.io/en/latest/notebooks/Common_Gotchas_in_JAX.html#control-flow
jit's static_argnums parameter	https://jax.readthedocs.io/en/latest/jax.html#just-in-time-compilation-jit
structured control flow primitives	https://jax.readthedocs.io/en/latest/jax.lax.html#control-flow-operators
lax.scan	https://jax.readthedocs.io/en/latest/_autosummary/jax.lax.scan.html#jax.lax.scan
	https://patch-diff.githubusercontent.com/sjtechdev/jax#installation
build JAX from source	https://jax.readthedocs.io/en/latest/developer.html#building-from-source
Windows Subsystem for Linux	https://docs.microsoft.com/en-us/windows/wsl/about
built from source for Windows	https://jax.readthedocs.io/en/latest/developer.html#additional-notes-for-building-jaxlib-from-source-on-windows
Issue #5795 thread	https://github.com/google/jax/issues/5795
	https://patch-diff.githubusercontent.com/sjtechdev/jax#pip-installation-cpu
above	https://patch-diff.githubusercontent.com/sjtechdev/jax#installation
	https://patch-diff.githubusercontent.com/sjtechdev/jax#pip-installation-gpu-cuda
CUDA	https://developer.nvidia.com/cuda-downloads
CuDNN	https://developer.nvidia.com/CUDNN
building from source	https://jax.readthedocs.io/en/latest/developer.html#building-from-source
CUDA toolkit's corresponding driver version	https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html#cuda-major-component-versions__table-cuda-toolkit-driver-versions
CUDA forward compatibility packages	https://docs.nvidia.com/deploy/cuda-compatibility/
above	https://patch-diff.githubusercontent.com/sjtechdev/jax#installation
the issue tracker	https://github.com/google/jax/issues
	https://patch-diff.githubusercontent.com/sjtechdev/jax#pip-installation-google-cloud-tpu
Google Cloud TPU	https://cloud.google.com/tpu/docs/users-guide-tpu-vm
	https://patch-diff.githubusercontent.com/sjtechdev/jax#pip-installation-colab-tpu
	https://patch-diff.githubusercontent.com/sjtechdev/jax#conda-installation
Tips & tricks	https://conda-forge.org/docs/user/tipsandtricks.html#installing-cuda-enabled-packages-like-tensorflow-and-pytorch
jaxlib	https://github.com/conda-forge/jaxlib-feedstock#installing-jaxlib
jax	https://github.com/conda-forge/jax-feedstock#installing-jax
	https://patch-diff.githubusercontent.com/sjtechdev/jax#building-jax-from-source
Building JAX from source	https://jax.readthedocs.io/en/latest/developer.html#building-from-source
	https://patch-diff.githubusercontent.com/sjtechdev/jax#neural-network-libraries
Flax	https://github.com/google/flax
ecosystem of libraries around JAX	https://deepmind.com/blog/article/using-jax-to-accelerate-our-research
Haiku	https://github.com/deepmind/dm-haiku
Optax	https://github.com/deepmind/optax
RLax	https://github.com/deepmind/rlax
chex	https://github.com/deepmind/chex
here	https://www.youtube.com/watch?v=iDxJxIyzSiM
	https://patch-diff.githubusercontent.com/sjtechdev/jax#citing-jax
jax/version.py	https://patch-diff.githubusercontent.com/sjtechdev/jax/blob/main/jax/version.py
paper that appeared at SysML 2018	https://mlsys.org/Conferences/2019/doc/2018/146.pdf
	https://patch-diff.githubusercontent.com/sjtechdev/jax#reference-documentation
reference documentation	https://jax.readthedocs.io/
developer documentation	https://jax.readthedocs.io/en/latest/developer.html
jax.readthedocs.io/	http://jax.readthedocs.io/
Readme	https://patch-diff.githubusercontent.com/sjtechdev/jax#readme-ov-file
Apache-2.0 license	https://patch-diff.githubusercontent.com/sjtechdev/jax#Apache-2.0-1-ov-file
Please reload this page	https://patch-diff.githubusercontent.com/sjtechdev/jax
Activity	https://patch-diff.githubusercontent.com/sjtechdev/jax/activity
0 stars	https://patch-diff.githubusercontent.com/sjtechdev/jax/stargazers
0 watching	https://patch-diff.githubusercontent.com/sjtechdev/jax/watchers
0 forks	https://patch-diff.githubusercontent.com/sjtechdev/jax/forks
Report repository	https://patch-diff.githubusercontent.com/contact/report-content?content_url=https%3A%2F%2Fgithub.com%2Fsjtechdev%2Fjax&report=sjtechdev+%28user%29
Releases	https://patch-diff.githubusercontent.com/sjtechdev/jax/releases
Packages 0	https://patch-diff.githubusercontent.com/users/sjtechdev/packages?repo_name=jax
	https://github.com
Terms	https://docs.github.com/site-policy/github-terms/github-terms-of-service
Privacy	https://docs.github.com/site-policy/privacy-policies/github-privacy-statement
Security	https://github.com/security
Status	https://www.githubstatus.com/
Community	https://github.community/
Docs	https://docs.github.com/
Contact	https://support.github.com?tags=dotcom-footer

Viewport: width=device-width

URLs of crawlers that visited me.