nn.modules.JaxModule ================================================================================================================================================================================================================================================== .. currentmodule:: nn.modules .. autoclass:: JaxModule :members: :private-members: :undoc-members: :inherited-members: :show-inheritance: .. rubric:: Attributes overview .. autosummary:: ~JaxModule.class_name ~JaxModule.full_name ~JaxModule.name ~JaxModule.shape ~JaxModule.size ~JaxModule.size_in ~JaxModule.size_out ~JaxModule.spiking_input ~JaxModule.spiking_output .. rubric:: Methods overview .. autosummary:: ~JaxModule.__init__ ~JaxModule.as_graph ~JaxModule.attributes_named ~JaxModule.evolve ~JaxModule.modules ~JaxModule.parameters ~JaxModule.reset_parameters ~JaxModule.reset_state ~JaxModule.set_attributes ~JaxModule.simulation_parameters ~JaxModule.state ~JaxModule.timed ~JaxModule.tree_flatten ~JaxModule.tree_unflatten