flax
Handle passing Variables as arguments in lifted transforms
#1397

Merged

Commits

Handle passing Variables and Module instances as arguments in lifted transforms

levskaya committed 4 years ago

Loading