PyTorch, TensorFlow, JAX and NumPy — all of them natively using the same code
MIT License
Bot releases are visible (Hide)
Published by jonasrauber over 4 years ago
ep.utils.get_dummy
TensorFlowTensor.sum
now automatically casts boolean to int (like PyTorchTensor
and others)Published by jonasrauber over 4 years ago
__mod__
ep.mean
(was sum
instead of mean
)Published by jonasrauber over 4 years ago
ep.astensor
and preform lazy imports
ep.torch
ep.tensorflow
ep.numpy
ep.jax
pow()
log_softmax()
kl_div_with_logits()
Published by jonasrauber almost 5 years ago
value_and_grad
(function and method)value_aux_and_grad
(function and method)value_and_grad_fn
(function)crossentropy
__repr__()
shown for EagerPy tensorsPublished by jonasrauber almost 5 years ago
__format__
support to tensorsfull
full_like
arange
index_update
cumsum
newaxis
flip
meshgrid
pad
nan
inf
isnan
isinf
Plus: Lot's of bug fixes and improvements.
Published by jonasrauber almost 5 years ago
softmax
stack
squeeze
and expand_dims
Published by jonasrauber almost 5 years ago
where
tile
matmul
logical_not
Published by jonasrauber almost 5 years ago
exp
, log
, log2
, log10
, log1p
Published by jonasrauber almost 5 years ago
bool()
all()
and any()
logical_and()
and logical_or()
Published by jonasrauber almost 5 years ago
concatenate
transpose
abs
Published by jonasrauber almost 5 years ago
onehot_like
, from_numpy
, ones
, zeros
, ones_like
, zeros_like
, __rsub__
__getitem__
support for TensorFlow (to match PyTorch and JAX)astensor
can now handle JAX arraysPublished by jonasrauber almost 5 years ago
JAXTensor
)Tensor.item()
Tensor.float32()
Tensor.__bool__()
istensor()
functionPublished by jonasrauber almost 5 years ago
eagerpy.astensor()
that automatically wraps it's input using the appropriate EagerPy tensorPublished by jonasrauber almost 5 years ago
PyTorchTensor
)TensorFlowTensor
)NumPyTensor
)