← all repositories

jonasrauber/eagerpy

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

697 stars Python ML Frameworks
eagerpy
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.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.