Peter Hawkins efefeac450 Drop flatbuffers as a Python dependency of JAX. ...

Change the pocketfft custom kernel in jaxlib to generate its flatbuffer descriptor in C++ instead. Surprisingly this code is actually much more readable in C++ because the flatbuffers Python API does not have a readable but less efficient API.

Breaking changes to the flatbuffers Python APIs have caused breakage in JAX in the past, and we can avoid the dependency completely without much work.

PiperOrigin-RevId: 457460347

2022-06-27 06:14:07 -07:00

..

rocm

feat: refactor code using pyupgrade

2022-05-17 22:14:05 +01:00

build_jaxlib_wheels_helpers.sh

Remove -it flag from docker run command in jaxlib build script.

2021-04-14 09:44:42 -07:00

build_jaxlib_wheels_macos.sh

Drop support for NumPy 1.18.

2022-01-04 12:11:38 -08:00

build_jaxlib_wheels.sh

Drop Python 3.6 support.

2021-07-15 14:20:29 -04:00

build_wheel_docker_entrypoint.sh

Allow manylinux_2_12_x86_64 in auditwheel test. This is an alias by PEP600.

2021-05-07 18:22:28 -04:00

build_wheel.py

Drop flatbuffers as a Python dependency of JAX.

2022-06-27 06:14:07 -07:00

BUILD.bazel

Add a build flag that allows disabling remote TPU builds.

2022-06-23 21:14:52 +00:00

build.py

Add a build flag that allows disabling remote TPU builds.

2022-06-23 21:14:52 +00:00

Dockerfile

Drop support for NumPy 1.18.

2022-01-04 12:11:38 -08:00

install_cuda.sh

Update cuDNN versions to highest available.

2021-03-09 19:07:00 +00:00

LICENSE.txt

Refactor JAX CPU kernels to make them usable from C++.

2021-09-03 10:03:54 -07:00

test-requirements.txt

Drop flatbuffers as a Python dependency of JAX.

2022-06-27 06:14:07 -07:00