jax
f0ab673d - Fully stage out scan under jit. (#2638)

Commit

6 years ago

Fully stage out scan under jit. (#2638) This prevents creating large arrays from the results of partial_eval.

References

Author

skye

Parents