jax authors 16b29a6930 Merge pull request #19288 from pearu:pearu/int32-overflow ...

PiperOrigin-RevId: 608701959

2024-02-20 12:43:16 -08:00

..

_ducc_fft.pyi

Add python stub files for jaxlib/cpu C++ Python extensions.

2023-11-28 08:45:24 -08:00

_lapack.pyi

Add nb::arg to nanobind definitions to generate better python annotations.

2023-11-30 10:39:28 -08:00

BUILD

Merge pull request #19288 from pearu:pearu/int32-overflow

2024-02-20 12:43:16 -08:00

cpu_kernels.cc

Add support for dynamic shapes to jax.fft.

2023-06-17 04:50:54 -07:00

ducc_fft_kernels.cc

Update ducc to commit: 2b2cead005e08d2632478e831d7f45da754162dc

2023-08-07 13:06:43 -07:00

ducc_fft_kernels.h

Add support for dynamic shapes to jax.fft.

2023-06-17 04:50:54 -07:00

ducc_fft.cc

Switch jaxlib to use nanobind instead of pybind11.

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

ducc_fft.fbs

Add support for dynamic shapes to jax.fft.

2023-06-17 04:50:54 -07:00

lapack_kernels_using_lapack.cc

Add support for Hessenberg and tridiagonal matrix reductions on CPU.

2022-11-09 06:23:55 -08:00

lapack_kernels.cc

Prevent silent overflow in lapack worker size calculations.

2024-02-20 11:04:06 +02:00

lapack_kernels.h

Switch JAX to use the OpenXLA repository.

2023-03-13 18:38:26 +00:00

lapack.cc

Add nb::arg to nanobind definitions to generate better python annotations.

2023-11-30 10:39:28 -08:00