Peter Hawkins 70b7d50181 Switch jaxlib to use nanobind instead of pybind11. ...

nanobind has a number of advantages (https://nanobind.readthedocs.io/en/latest/why.html), notably speed of compilation and dispatch, but the main reason to do this for these bindings is because nanobind can target the Python Stable ABI starting with Python 3.12. This means that we will not need to ship per-Python version CUDA plugins starting with Python 3.12.

PiperOrigin-RevId: 559898790

2023-08-24 16:07:56 -07:00

..

jax_cpp

Switch jaxlib to use nanobind instead of pybind11.

2023-08-24 16:07:56 -07:00

__init__.py

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

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

advi.py

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

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

datasets.py

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

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

differentially_private_sgd.py

Update flags to use the ABSL typed flag API.

2023-07-27 12:15:58 -07:00

examples_test.py

Change to simpler import for jax.config

2023-04-21 11:51:22 -07:00

gaussian_process_regression.py

Fix for log-normal prior in example.

2023-08-09 13:38:37 +01:00

kernel_lsq.py

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

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

mnist_classifier_fromscratch.py

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

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

mnist_classifier.py

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

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

mnist_vae.py

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

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

onnx2xla.py

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

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

spmd_mnist_classifier_fromscratch.py

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

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