jonasrauber/eagerpy
A Python library providing a unified API across PyTorch, TensorFlow, JAX, and NumPy for writing framework-agnostic tensor operations.

Velocity · 7d
+0.3
★ / day
Trend
→steady
star history
EagerPy is a Python framework that enables writing code that automatically works natively with PyTorch, TensorFlow, JAX, and NumPy. It provides fully chainable methods on tensor objects and extensive type annotations. Operations are translated directly into native operations of the underlying framework, ensuring performance while offering a consistent API across backends.