Kanglan Tang af28595909 Add a jax_wheel Bazel rule to build jax pip packages ...

PiperOrigin-RevId: 689514531

2024-10-24 14:20:46 -07:00

..

toolchains

Add a config for using Clang on Windows.

2024-05-06 10:39:28 -07:00

__init__.py

Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".

2022-09-22 12:27:19 -07:00

BUILD

Fix some load orderings for buildifier

2024-03-27 10:28:57 -07:00

build_defs.bzl

Clean up BUILD files.

2024-08-26 09:11:17 -07:00

build_utils.py

Add a jax_wheel Bazel rule to build jax pip packages

2024-10-24 14:20:46 -07:00

colab_tpu.py

Use shorter error message for jax.tools.colab_tpu.setup_tpu()

2024-03-05 11:20:38 -08:00

jax_to_ir.py

[jax2tf] Remove non-native serialization test from jax_to_ir_test

2024-10-07 04:21:38 -07:00

pgo_nsys_converter.py

fixes for mypy

2024-10-02 13:46:10 -07:00