v0.8.0
Breaking changes
LazyArray.__iter__: now iterates over slices of array rather than the computational graph nodes
Enhancements
- add
xp = autoray.get_namespace(like)as an alternative api LazyArray: add support for python array api via.__array_namespace__()LazyArray: support broadcasted linear algebra- alias
("torch" "equal")totorch.eq lazy: support caching of more kwargslazy: addtake_along_axislazy: addequal- add
"random.default_rng"implementation forjaxandtorchto support random pure functions (#27)
Bug fixes
- python compiler supports calls with kwargs
jax: re wrap fat QR to support gradientpytensor: qr fix mode to 'economic'