Source-built JAX frontend: differentiate, compile, and transform Numpy code.
This is the JAX frontend package, it's meant to be used together with one of the jaxlib implementations, e.g. python3Packages.jaxlib, python3Packages.jaxlib-bin, or python3Packages.jaxlibWithCuda.
0.5.0aarch64-darwinaarch64-freebsdaarch64-linuxaarch64-windowsarmv5tel-linuxarmv6l-linuxarmv7a-darwinarmv7a-linuxarmv7l-linuxi686-cygwini686-darwini686-freebsdi686-linuxi686-windowsloongarch64-linuxm68k-linuxmicroblaze-linuxmicroblazeel-linuxmips-linuxmips64-linuxmips64el-linuxmipsel-linuxpowerpc64-linuxpowerpc64le-linuxriscv32-linuxriscv64-linuxs390-linuxs390x-linuxx86_64-cygwinx86_64-darwinx86_64-freebsdx86_64-linuxx86_64-windows