nn.modules.RateJax
==================================================================================================================================================================================================================================================

.. currentmodule:: nn.modules

.. autoclass:: RateJax
   :members:
   :private-members:
   :undoc-members:
   :inherited-members:
   :show-inheritance:

   
   
   .. rubric:: Attributes overview

   .. autosummary::
   
      ~RateJax.class_name
      ~RateJax.full_name
      ~RateJax.name
      ~RateJax.shape
      ~RateJax.size
      ~RateJax.size_in
      ~RateJax.size_out
      ~RateJax.spiking_input
      ~RateJax.spiking_output
      ~RateJax.rng_key
      ~RateJax.x
      ~RateJax.w_rec
      ~RateJax.tau
      ~RateJax.bias
      ~RateJax.threshold
      ~RateJax.dt
      ~RateJax.noise_std
      ~RateJax.act_fn
   
   


   

   
   .. rubric:: Methods overview

   .. autosummary::
   
      ~RateJax.__init__
      ~RateJax.as_graph
      ~RateJax.attributes_named
      ~RateJax.evolve
      ~RateJax.modules
      ~RateJax.parameters
      ~RateJax.reset_parameters
      ~RateJax.reset_state
      ~RateJax.set_attributes
      ~RateJax.simulation_parameters
      ~RateJax.state
      ~RateJax.timed
      ~RateJax.tree_flatten
      ~RateJax.tree_unflatten