v0.3.7

New Features:

• nt.stax.Cos
• nt.stax.ImageResize
• New implementation implementation="SPARSE" in nt.stax.Aggregate for efficient handling of sparse graphs (see #86, #9)
• Support approximate=True in nt.stax.Gelu

Bugfixes:

• Fix a bug that might alter Kernel requirements
• Fix nt.batch handling of diagonal_axes (see #87)
• Remove the frequent but redundant warning about type conversion in kernel_fn
• Minor fixes to documentation and code clean-up

Breaking changes:

• Parameters initialized by init_fn now follow the setting of JAX_ENABLE_X64 instead of always defaulting to 32-bit (see #112)
• Drop python 3.6 support and add python 3.9 support