jax
e6854009 - Changed axis kwarg in wrapper for py2 compat.

Commit
7 years ago
Changed axis kwarg in wrapper for py2 compat.
Author
Alex Alemi
Parents
Loading