eagerpy (pypi): EagerPy is a thin wrapper around PyTorch, TensorFlow Eager, JAX and NumPy that unifies their interface and thus allows writing code that works natively across all of them.
Keywords: eager-execution|jax|numpy|python|pytorch|tensorflow|tensorflow2