jax
1b597895 - add ShardedDeviceArray to ad vspace op handlers

Commit

6 years ago

add ShardedDeviceArray to ad vspace op handlers fixes #2529 (thanks, @dpfau !)

References

#2532 - add ShardedDeviceArray to ad vspace op handlers

Author

mattjj

mattjj

Parents

Loading