v2.0.1

Full Changelog: v2.0.0...v2.0.1

New features:

• Add zero_param_mask that strengthens juice.blockify
• reuse parameter flows space for tied nodes in different layers

Bug fixes:

• fix nan error when nmars contains -inf
• clean up & fix flow assigning kernel
• sum layer large sparse forward: handle cases with very high grid size
• fix source ns _param_range assignment error
• fix compilation logic for tied nodes when the source does not appear first
• fix bug: set appropriate other values in pflow accumulation kernels
• fix: use_block_sparse_edges does not set to False when only 1 ProdNodes in a ProdLayer

API:

• accept 2d sum params for fully connected nodes
• change from "group" to "block"