Adapted jax_ops.py
to handle the lowering interfaces changed from jax v0.4.16 onwards.
#39
Annotations
1 warning
Run docker/setup-qemu-action@v3
Unexpected input(s) 'path', 'name', valid inputs are ['image', 'platforms']
|
Loading