jax-ml/jax - Morph

SIGN IN SIGN UP

jax-ml / jax UNCLAIMED

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

35263 0 0 Python

50 branches 275 tags

jax authors Merge pull request #36346 from mattjj:remove-split-transpose

eb95074 · 13h ago · 34,628 Commits

	.github
	benchmarks
	build
	ci
	cloud_tpu_colabs
	docs
	examples
	images
	jax
	jax_plugins
	jaxlib
	tests
	third_party
	.bazelrc		26.1 KB
	.bazelversion		6 B
	.editorconfig		176 B
	.gitignore		399 B
	.pre-commit-config.yaml		1.9 KB
	.readthedocs.yml		1.1 KB
	AUTHORS		313 B
	build_wheel.py		3.3 KB
	BUILD.bazel		6.1 KB
	CHANGELOG.md		182.0 KB
	CITATION.bib		408 B
	conftest.py		3.5 KB
	CONTRIBUTING.md		144 B
	LICENSE		11.1 KB
	MODULE.bazel		9.0 KB
	platform_mappings		371 B
	pyproject.toml		4.3 KB
	README.md		10.6 KB
	setup.py		5.0 KB
	test_shard_count.bzl		1.0 KB
	WORKSPACE		6.1 KB