JAX

“Making Keras multi-backend again in 2023 has been one of my favorite projects to work on, both from the engineering & architecture side”

“To me this is the real power of JAX, it can be viewed as a few functions that make it easy to take any python code”

“I highly recommend JAX to power users. It's nowhere near as feature-complete from a neural network sense as, say, PyTorch, but it is very good at what it does”