Submissions from jax.readthedocs.io

		Pallas: A Jax Kernel Language (jax.readthedocs.io)
		3 points by lnyan 3 months ago \| past
		Pallas: A Jax Kernel Language (jax.readthedocs.io)
		2 points by nphard85 5 months ago \| past
		JAX – NumPy on the CPU, GPU, and TPU (jax.readthedocs.io)
		276 points by peter_d_sherman 9 months ago \| past \| 143 comments
		Pytrees (jax.readthedocs.io)
		132 points by f_devd on May 22, 2023 \| past \| 14 comments
		Autodidax: JAX core from scratch (jax.readthedocs.io)
		107 points by sva_ on Feb 9, 2023 \| past \| 12 comments
		Autodidax: Jax Core from Scratch (jax.readthedocs.io)
		1 point by iflp on Sept 1, 2022 \| past
		Autodidax (jax.readthedocs.io)
		1 point by mooreds on March 6, 2022 \| past
		Autodiff Cookbook: Part 1 – easy and powerful automatic differentiation in JAX (jax.readthedocs.io)
		2 points by sebg on Aug 21, 2020 \| past