keras

[jax]
jax
jaxlib

[tensorflow]
tensorflow
