jax_md_mod.model.layers.SphericalBesselLayer.params_dict

SphericalBesselLayer.params_dict()

Returns parameters keyed by name for this module and submodules.

Return type:

Mapping[str, Array]