Releases: jcmgray/autoray
Releases · jcmgray/autoray
v0.8.10
v0.8.9
v0.8.8
v0.8.7
Enhancements
- Torch improvements:
- Added torch implementation for
tracewith axes specified plus tests. - Allowed torch
default_rngseed to be None. - Added axis kwarg support for torch
count_nonzero.
- Added torch implementation for
- Added decorator support for registration APIs:
register_functioncan now be used as a decorator.register_custom_wrappercan now be used as a decorator.
- Refactored internals to use the new decorator style for function registration.
v0.8.6
What's Changed
- add stop_gradient by @ChenAo-Phys in #29
- Add torch support for scipy.linalg.solve_triangular.
- Add tensorflow support for random.default_rng.
- Add allclose support for lazy arrays.
- Add Python 3.14 support.
- Fix and sort internal backend registration functions.
New Contributors
- @ChenAo-Phys made their first contribution in #29
Full Changelog: v0.8.5...v0.8.6
v0.8.4
Enhancements
- add
autoray.is_scalarfunction lazy.einsum: for numpy backend setoptimize=Trueby defaultlazy.linalg.norm: add kwargs supporttorch.random.default_rng: add choice support- move ci to pixi
Full Changelog: v0.8.3...v0.8.4
v0.8.3
Enhancements
- register
arrayandasarrayas creation routines, so that they pick up device by default fromlike LazyArray.show: show more information about call signature by default- lazy creation routines, support single int as 1D shape specifier
- remove custom
torch.count_nonzerowhichtorchnow natively implements - bump min python version to 3.10
torch.takeimplementation: support vmapping over both scalar and sequence indicesget_namespace: supportscipyas a custom submodule.- custom
autograd.numpy.taketo support grad (HIPS/autograd#743).
Full Changelog: v0.8.2...v0.8.3
v0.8.2
Ehancements
- Speedup creation of
get_namespace. - Reduce namespace attribute retrieval overhead to essentially zero
Full Changelog: v0.8.1...v0.8.2
v0.8.1
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'