Index

_

__call__() (neurai.nn.layer.conv.ConvTranspose method)

A

Acitivate (class in neurai.nn.layer.activate) activate_fun (neurai.nn.layer.activate.Relu attribute) (neurai.nn.layer.activate.Relu6 attribute) activate_fun() (neurai.nn.layer.activate.Celu method) (neurai.nn.layer.activate.Elu method) (neurai.nn.layer.activate.Gelu method) (neurai.nn.layer.activate.Glu method) (neurai.nn.layer.activate.HardSigmoid method) (neurai.nn.layer.activate.HardSilu method) (neurai.nn.layer.activate.HardSwish method) (neurai.nn.layer.activate.HardTanh method) (neurai.nn.layer.activate.LeakyRelu method) (neurai.nn.layer.activate.LogSigmoid method) (neurai.nn.layer.activate.LogSoftmax method) (neurai.nn.layer.activate.LogSumexp method) (neurai.nn.layer.activate.Selu method) (neurai.nn.layer.activate.Sigmoid method) (neurai.nn.layer.activate.Silu method) (neurai.nn.layer.activate.Softmax method) (neurai.nn.layer.activate.Softplus method) (neurai.nn.layer.activate.SoftSign method) (neurai.nn.layer.activate.Tanh method)

AdaptiveRungekutta (class in neurai.math.ode.runge_kutta) adjust() (neurai.mp.scale.DynamicLossScale method) (neurai.mp.scale.NoOpLossScale method) (neurai.mp.scale.StaticLossScale method) ALIFCell (class in neurai.nn.layer.recurrent) All2All (class in neurai.nn.conn.connrule) all_dt() (neurai.nn.neuron.hh.HH method) (neurai.nn.neuron.hh_psc.HHPSC method) all_finite() (in module neurai.mp.scale) AMPA (class in neurai.nn.neuron.multi_compartment.compartment) apply_updates() (in module neurai.opt.update) attend() (neurai.nn.layer.attention.Embed method) AvgPool (class in neurai.nn.layer.pooling)

B

BatchNorm (class in neurai.nn.layer.normalization) BatchNorm1d (class in neurai.nn.layer.normalization) BatchNorm2d (class in neurai.nn.layer.normalization) BatchNorm3d (class in neurai.nn.layer.normalization) BernoulliGenerator (class in neurai.nn.neuron.bernoulli_generator)

BernoulliIniter (class in neurai.initializer.initializer) bias_init() (neurai.nn.layer.pooling.SNNPool method) binary_cross_entropy() (in module neurai.nn.loss) BiPC (class in neurai.grads.autograd) BP (class in neurai.grads.autograd) build_network_from_json() (in module neurai.util.ir_collector)

C

CallbackRecorder (class in neurai.monitor) canonicalize_dtype() (in module neurai.nn.layer.attention) canonicalize_padding() (in module neurai.nn.layer.conv) cast_to_compute() (neurai.mp.policy.Policy method) cast_to_full() (in module neurai.mp.policy) cast_to_half() (in module neurai.mp.policy) cast_to_output() (neurai.mp.policy.Policy method) cast_to_param() (neurai.mp.policy.Policy method) Celu (class in neurai.nn.layer.activate) check_kernel_dimensions() (neurai.nn.layer.conv.Conv method) Compartment (class in neurai.nn.neuron.multi_compartment.compartment) ComputationType (class in neurai.const.train) Connector (class in neurai.nn.conn.connrule) ConnectRepr (class in neurai.const.simulation) ConnGen (class in neurai.nn.neuron.multi_compartment.compartment)

ConstantIniter (class in neurai.initializer.initializer) ConstantPad (class in neurai.nn.layer.padding) Conv (class in neurai.nn.layer.conv) Conv1d (class in neurai.nn.layer.conv) Conv2d (class in neurai.nn.layer.conv) Conv3d (class in neurai.nn.layer.conv) convert2nest_spike() (in module neurai.util.write) convert2nest_voltage() (in module neurai.util.write) ConvTranspose (class in neurai.nn.layer.conv) ConvTranspose1d (class in neurai.nn.layer.conv) ConvTranspose2d (class in neurai.nn.layer.conv) ConvTranspose3d (class in neurai.nn.layer.conv) CUDA (neurai.const.train.ComputationType attribute) CustomConn (class in neurai.nn.conn.connrule) CustomError CustomGenerator (class in neurai.nn.neuron.custom_generator)

D

Dataset (class in neurai.datasets.base) DCGenerator (class in neurai.nn.neuron.dc_generator) DebugWriteFormatChecker (class in neurai.util.write) default() (neurai.util.ir_collector.NpEncoder method) DelaySpike (class in neurai.nn.delay_spike)

diff_all2all() (in module neurai.ops.target_vector_operators) diff_coo() (in module neurai.ops.target_vector_operators) diff_mat() (in module neurai.ops.target_vector_operators) diff_one2one() (in module neurai.ops.target_vector_operators) Dropout (class in neurai.nn.layer.dropout) DynamicLossScale (class in neurai.mp.scale)

E

Elu (class in neurai.nn.layer.activate) Embed (class in neurai.nn.layer.attention) EP (class in neurai.grads.autograd) EPMode (class in neurai.const.train) ErrorCode (class in neurai.const.error) Euler (class in neurai.math.ode.euler) EULER (neurai.const.math.SolverMethodODE attribute) Event (class in neurai.monitor) event_all2all() (in module neurai.ops.target_vector_operators) event_coo() (in module neurai.ops.target_vector_operators)

event_csr() (in module neurai.ops.target_vector_operators) event_mat() (in module neurai.ops.target_vector_operators) event_one2one() (in module neurai.ops.target_vector_operators) Exodus (class in neurai.nn.layer.exodus) exodus_cpu_bwd() (in module neurai.nn.layer.exodus) exodus_cpu_fwd() (in module neurai.nn.layer.exodus) EXP_EULER (neurai.const.math.SolverMethodODE attribute) ExplictRungeKutta (class in neurai.math.ode.runge_kutta) ExpLIF (class in neurai.nn.neuron.exp_lif) ExponentionalEuler (class in neurai.math.ode.exp_euler)

F

FileRecorder (class in neurai.monitor) FixedIndegree (class in neurai.nn.conn.connrule) FixedOutdegree (class in neurai.nn.conn.connrule) FixedProb (class in neurai.nn.conn.connrule) FixedTotalNum (class in neurai.nn.conn.connrule)

Flatten (class in neurai.nn.layer.linear) flatten() (in module neurai.nn.functional) FORLOOP (neurai.const.train.ComputationType attribute) func() (neurai.nn.layer.pooling.AvgPool method) (neurai.nn.layer.pooling.MaxPool method) (neurai.nn.layer.pooling.MinPool method)

G

Gaussian (class in neurai.grads.surrogate_grad) Gelu (class in neurai.nn.layer.activate) Generator (class in neurai.nn.neuron.generator) get_alpha() (neurai.nn.layer.exodus.Exodus method) get_dendritic_delay_data() (neurai.nn.neuron.neuron.Neuron method) get_dendritic_delay_trace() (neurai.nn.neuron.neuron.Neuron method) get_dendritic_delay_trace_triplet() (neurai.nn.neuron.neuron.Neuron method) get_grads() (neurai.grads.autograd.BiPC method) (neurai.grads.autograd.BP method) (neurai.grads.autograd.EP method) (neurai.grads.autograd.PC method)

get_policy() (in module neurai.mp.policy) GLIF3 (class in neurai.nn.neuron.glif) GLIF3PSC (class in neurai.nn.neuron.glif_psc) Glu (class in neurai.nn.layer.activate) grad() (in module neurai.grads.autograd) GradeBase (class in neurai.grads.autograd) GRUCell (class in neurai.nn.layer.recurrent)

H

half_dtype() (in module neurai.mp.policy) HARD (neurai.const.reset_type.ResetType attribute) HardSigmoid (class in neurai.nn.layer.activate) HardSilu (class in neurai.nn.layer.activate) HardSwish (class in neurai.nn.layer.activate) HardTanh (class in neurai.nn.layer.activate)

Heun (class in neurai.math.sde.heun) Heun2 (class in neurai.math.ode.heun) Heun3 (class in neurai.math.ode.heun) HH (class in neurai.nn.neuron.hh) HHPSC (class in neurai.nn.neuron.hh_psc) hinge_loss() (in module neurai.nn.loss) huber_loss() (in module neurai.nn.loss)

I

IMPLICT_EULER (neurai.const.math.SolverMethodODE attribute) ImplictEuler (class in neurai.math.ode.implict_euler) init_neuron_state() (neurai.nn.layer.exodus.Exodus method) (neurai.nn.neuron.neuron.StepwiseNeuron method) (neurai.nn.neuron.snn_alif.SNNALIF method) (neurai.nn.neuron.snn_lif.SparseLIF method) (neurai.nn.neuron.snn_paramtriclif.SNNParamtricLIF method) initial_state() (neurai.nn.layer.recurrent.GRUCell method) (neurai.nn.layer.recurrent.LSTMCell method) (neurai.nn.layer.recurrent.RNNCellBase method) (neurai.nn.layer.recurrent.SRNNCellBase method)

Initializer (class in neurai.initializer.initializer) InputTransmitter (class in neurai.nn.neuron.input_transmitter) interpolate() (in module neurai.nn.functional) is_prompt_learning (neurai.finetune.finetune_config.LoRAConfig property) (neurai.finetune.finetune_config.PromptLearningConfig property) Izhikevich (class in neurai.nn.neuron.izhikevich)

J

jit() (in module neurai.util.trans)

K

KaimingUniformIniter (class in neurai.initializer.initializer)

kernel_init() (neurai.nn.layer.pooling.SNNPool method)

L

l1_loss() (in module neurai.nn.loss) LayerNorm (class in neurai.nn.layer.normalization) LeakyIF (class in neurai.nn.neuron.multi_compartment.compartment) LeakyRelu (class in neurai.nn.layer.activate) LecunNormalIniter (class in neurai.initializer.initializer) LIF (class in neurai.nn.neuron.lif) LIFCell (class in neurai.nn.layer.recurrent) line_plot() (in module neurai.util.visualization)

Linear (class in neurai.nn.layer.linear) load_data() (in module neurai.util.write) LocalBase (class in neurai.grads.autograd) LogSigmoid (class in neurai.nn.layer.activate) LogSoftmax (class in neurai.nn.layer.activate) LogSumexp (class in neurai.nn.layer.activate) LoRAConfig (class in neurai.finetune.finetune_config) LoraWeight (class in neurai.finetune.tuning.lorax.transform) LSTMCell (class in neurai.nn.layer.recurrent)

M

masked_softmax() (in module neurai.nn.layer.attention) MaxPool (class in neurai.nn.layer.pooling) maybe_replicate() (in module neurai.nn.layer.conv) MCConn (class in neurai.nn.neuron.multi_compartment.compartment) MCConnSynapse (class in neurai.nn.neuron.multi_compartment.compartment) MCNeuron (class in neurai.nn.neuron.multi_compartment.mc_neuron) MCSynapse (class in neurai.nn.neuron.multi_compartment.compartment) MemoryRecorder (class in neurai.monitor) MidPoint (class in neurai.math.ode.midpoint) MinPool (class in neurai.nn.layer.pooling) Mish (class in neurai.nn.layer.activate) Mislteain (class in neurai.math.sde.mislteain) module neurai neurai.config neurai.const neurai.const.error neurai.const.math neurai.const.reset_type neurai.const.simulation neurai.const.train neurai.datasets.base neurai.finetune.finetune_config neurai.finetune.tuning.lorax neurai.finetune.tuning.lorax.constants neurai.finetune.tuning.lorax.helpers neurai.finetune.tuning.lorax.transform neurai.grads neurai.grads.autograd neurai.grads.surrogate_grad neurai.initializer neurai.initializer.initializer neurai.manager neurai.manager.ide_manager neurai.manager.rng_manager neurai.manager.var_manager neurai.math neurai.math.ode neurai.math.ode.euler neurai.math.ode.exp_euler neurai.math.ode.heun neurai.math.ode.implict_euler neurai.math.ode.midpoint neurai.math.ode.ode neurai.math.ode.ralston neurai.math.ode.rk neurai.math.ode.runge_kutta neurai.math.sde neurai.math.sde.heun neurai.math.sde.mislteain neurai.math.sde.sde neurai.math.solver neurai.metrics neurai.metrics.snn_metrics neurai.monitor neurai.mp neurai.mp.policy neurai.mp.scale neurai.nn.conn neurai.nn.conn.connrule neurai.nn.delay_spike neurai.nn.functional neurai.nn.kwonly_dataclasses neurai.nn.layer.activate neurai.nn.layer.attention neurai.nn.layer.conv neurai.nn.layer.dropout neurai.nn.layer.exodus neurai.nn.layer.linear neurai.nn.layer.normalization neurai.nn.layer.padding neurai.nn.layer.pooling neurai.nn.layer.recurrent neurai.nn.layer.slayer neurai.nn.loss neurai.nn.module neurai.nn.neuron neurai.nn.neuron.bernoulli_generator neurai.nn.neuron.custom_generator neurai.nn.neuron.dc_generator neurai.nn.neuron.exp_lif neurai.nn.neuron.generator neurai.nn.neuron.glif neurai.nn.neuron.glif_psc neurai.nn.neuron.hh neurai.nn.neuron.hh_psc neurai.nn.neuron.input_transmitter neurai.nn.neuron.izhikevich neurai.nn.neuron.lif neurai.nn.neuron.multi_compartment neurai.nn.neuron.multi_compartment.compartment neurai.nn.neuron.multi_compartment.mc_neuron neurai.nn.neuron.neuron neurai.nn.neuron.poisson_generator neurai.nn.neuron.snn_alif neurai.nn.neuron.snn_eif neurai.nn.neuron.snn_if neurai.nn.neuron.snn_klif neurai.nn.neuron.snn_lif neurai.nn.neuron.snn_paramtriclif neurai.nn.neuron.snn_qif neurai.nn.neuron.snn_srm neurai.nn.neuron.spike_generator neurai.ops neurai.ops.target_vector_operators neurai.opt neurai.opt.update neurai.setting neurai.util neurai.util.apu neurai.util.dimension_definitions neurai.util.doc neurai.util.init_func neurai.util.ir_collector neurai.util.ir_to_net neurai.util.mpi_init neurai.util.res_collector neurai.util.serialization neurai.util.sim_build neurai.util.trans neurai.util.visualization neurai.util.write neurai.version

Monitor (class in neurai.monitor) MonitorBS (class in neurai.monitor) MonitorConf (class in neurai.monitor) mse_loss() (in module neurai.nn.loss) MULTI (neurai.const.train.StepMode attribute) MultiGaussian (class in neurai.grads.surrogate_grad)

N

neurai module neurai.config module neurai.const module neurai.const.error module neurai.const.math module neurai.const.reset_type module neurai.const.simulation module neurai.const.train module neurai.datasets.base module neurai.finetune.finetune_config module neurai.finetune.tuning.lorax module neurai.finetune.tuning.lorax.constants module neurai.finetune.tuning.lorax.helpers module neurai.finetune.tuning.lorax.transform module neurai.grads module neurai.grads.autograd module neurai.grads.surrogate_grad module neurai.initializer module neurai.initializer.initializer module neurai.manager module neurai.manager.ide_manager module neurai.manager.rng_manager module neurai.manager.var_manager module neurai.math module neurai.math.ode module neurai.math.ode.euler module neurai.math.ode.exp_euler module neurai.math.ode.heun module neurai.math.ode.implict_euler module neurai.math.ode.midpoint module neurai.math.ode.ode module neurai.math.ode.ralston module neurai.math.ode.rk module neurai.math.ode.runge_kutta module neurai.math.sde module neurai.math.sde.heun module neurai.math.sde.mislteain module neurai.math.sde.sde module neurai.math.solver module neurai.metrics module neurai.metrics.snn_metrics module neurai.monitor module neurai.mp module neurai.mp.policy module neurai.mp.scale module neurai.nn.conn module neurai.nn.conn.connrule module neurai.nn.delay_spike module neurai.nn.functional module neurai.nn.kwonly_dataclasses module neurai.nn.layer.activate module neurai.nn.layer.attention module neurai.nn.layer.conv module neurai.nn.layer.dropout module neurai.nn.layer.exodus module neurai.nn.layer.linear module neurai.nn.layer.normalization module neurai.nn.layer.padding module neurai.nn.layer.pooling module

neurai.nn.layer.recurrent module neurai.nn.layer.slayer module neurai.nn.loss module neurai.nn.module module neurai.nn.neuron module neurai.nn.neuron.bernoulli_generator module neurai.nn.neuron.custom_generator module neurai.nn.neuron.dc_generator module neurai.nn.neuron.exp_lif module neurai.nn.neuron.generator module neurai.nn.neuron.glif module neurai.nn.neuron.glif_psc module neurai.nn.neuron.hh module neurai.nn.neuron.hh_psc module neurai.nn.neuron.input_transmitter module neurai.nn.neuron.izhikevich module neurai.nn.neuron.lif module neurai.nn.neuron.multi_compartment module neurai.nn.neuron.multi_compartment.compartment module neurai.nn.neuron.multi_compartment.mc_neuron module neurai.nn.neuron.neuron module neurai.nn.neuron.poisson_generator module neurai.nn.neuron.snn_alif module neurai.nn.neuron.snn_eif module neurai.nn.neuron.snn_if module neurai.nn.neuron.snn_klif module neurai.nn.neuron.snn_lif module neurai.nn.neuron.snn_paramtriclif module neurai.nn.neuron.snn_qif module neurai.nn.neuron.snn_srm module neurai.nn.neuron.spike_generator module neurai.ops module neurai.ops.target_vector_operators module neurai.opt module neurai.opt.update module neurai.setting module neurai.util module neurai.util.apu module neurai.util.dimension_definitions module neurai.util.doc module neurai.util.init_func module neurai.util.ir_collector module neurai.util.ir_to_net module neurai.util.mpi_init module neurai.util.res_collector module neurai.util.serialization module neurai.util.sim_build module neurai.util.trans module neurai.util.visualization module neurai.util.write module neurai.version module Neuron (class in neurai.nn.neuron.neuron) neuronal_integrate() (neurai.nn.neuron.snn_eif.SNNEIF method) (neurai.nn.neuron.snn_if.SNNIF method) (neurai.nn.neuron.snn_klif.SNNKLIF method) (neurai.nn.neuron.snn_lif.SNNLIF method) (neurai.nn.neuron.snn_qif.SNNQIF method) neuronal_reset() (neurai.nn.neuron.neuron.StepwiseNeuron method) (neurai.nn.neuron.snn_klif.SNNKLIF method) NMDA (class in neurai.nn.neuron.multi_compartment.compartment) NoOpLossScale (class in neurai.mp.scale) norm_except_dim() (neurai.nn.layer.normalization.WeightNorm static method) normal_fun() (neurai.grads.autograd.EP method) NormalClipIniter (class in neurai.initializer.initializer) NormalIniter (class in neurai.initializer.initializer) NpEncoder (class in neurai.util.ir_collector) numSpikes() (neurai.nn.loss.SpikeLoss method)

O

ode_solve() (in module neurai.math.ode.ode)

ODESolver (class in neurai.math.ode.ode) One2One (class in neurai.nn.conn.connrule)

P

parse_dtype() (in module neurai.mp.policy) PC (class in neurai.grads.autograd) PCMode (class in neurai.const.train) PoissonGenerator (class in neurai.nn.neuron.poisson_generator) PoissonIniter (class in neurai.initializer.initializer) Policy (class in neurai.mp.policy) Polynomial (class in neurai.grads.surrogate_grad) Pool (class in neurai.nn.layer.pooling)

predict() (neurai.grads.autograd.BP method) (neurai.grads.autograd.EP method) (neurai.grads.autograd.LocalBase method) PrefixTuningConfig (class in neurai.finetune.finetune_config) ProgressData (class in neurai.manager.ide_manager) promote_dtype() (in module neurai.nn.layer.attention) PromptLearningConfig (class in neurai.finetune.finetune_config) psp_forward_jvp() (in module neurai.nn.layer.slayer)

R

Ralston2 (class in neurai.math.ode.ralston) Ralston3 (class in neurai.math.ode.ralston) Ralston4 (class in neurai.math.ode.ralston) raster_plot() (in module neurai.util.visualization) Recorder (class in neurai.monitor) Rectangular (class in neurai.grads.surrogate_grad) ReflectionPad (class in neurai.nn.layer.padding) refractory_kernel() (neurai.nn.layer.slayer.Slayer method) (neurai.nn.neuron.snn_srm.SNNSRM method) register_empty_pytree() (in module neurai.mp.scale) register_ode_solver() (in module neurai.math.ode.ode) register_sde_solver() (in module neurai.math.sde.sde) Relu (class in neurai.nn.layer.activate) Relu6 (class in neurai.nn.layer.activate) ReplicationPad (class in neurai.nn.layer.padding) reset() (neurai.nn.delay_spike.DelaySpike method) (neurai.nn.neuron.bernoulli_generator.BernoulliGenerator method) (neurai.nn.neuron.dc_generator.DCGenerator method) (neurai.nn.neuron.input_transmitter.InputTransmitter method) (neurai.nn.neuron.poisson_generator.PoissonGenerator method) (neurai.nn.neuron.spike_generator.SpikeGenerator method)

ResetType (class in neurai.const.reset_type) response_kernel() (neurai.nn.layer.slayer.Slayer method) (neurai.nn.neuron.snn_srm.SNNSRM method) restore() (in module neurai.util.serialization) RK2 (class in neurai.math.ode.rk) (neurai.const.math.SolverMethodODE attribute) RK3 (class in neurai.math.ode.rk) (neurai.const.math.SolverMethodODE attribute) RK4 (class in neurai.math.ode.rk) (neurai.const.math.SolverMethodODE attribute) RKF45 (class in neurai.math.ode.rk) (neurai.const.math.SolverMethodODE attribute) RngManager (class in neurai.manager.rng_manager) RNN (class in neurai.nn.layer.recurrent) RNNCellBase (class in neurai.nn.layer.recurrent)

S

save() (in module neurai.util.serialization) scale() (neurai.mp.scale.DynamicLossScale method) (neurai.mp.scale.NoOpLossScale method) (neurai.mp.scale.StaticLossScale method) sde_solve() (in module neurai.math.sde.sde) SDESolver (class in neurai.math.sde.sde) select_tree() (in module neurai.mp.scale) Selu (class in neurai.nn.layer.activate) sequence_mask() (in module neurai.nn.layer.attention) Sequential (class in neurai.nn.layer.linear) set_platform() (in module neurai.config) set_simulate_status() (in module neurai.config) SetupState (class in neurai.nn.module) Sigmoid (class in neurai.grads.surrogate_grad) (class in neurai.nn.layer.activate) sigmoid_binary_cross_entropy() (in module neurai.nn.loss) Silu (class in neurai.nn.layer.activate) simulate_init() (neurai.monitor.MonitorBS method) SINGLE (neurai.const.train.StepMode attribute) SingleExponential (class in neurai.grads.surrogate_grad) Slayer (class in neurai.nn.layer.slayer) SlayerPdf (class in neurai.grads.surrogate_grad) smooth_l1_loss() (in module neurai.nn.loss) SNNALIF (class in neurai.nn.neuron.snn_alif) SNNConv2d (class in neurai.nn.layer.conv) SNNConv3d (class in neurai.nn.layer.conv) SNNConvTranspose3d (class in neurai.nn.layer.conv) SNNDropout3d (class in neurai.nn.layer.dropout) SNNEIF (class in neurai.nn.neuron.snn_eif) SNNIF (class in neurai.nn.neuron.snn_if) SNNKLIF (class in neurai.nn.neuron.snn_klif)

SNNLIF (class in neurai.nn.neuron.snn_lif) SNNLinear3d (class in neurai.nn.layer.linear) SNNParamtricLIF (class in neurai.nn.neuron.snn_paramtriclif) SNNPool (class in neurai.nn.layer.pooling) SNNQIF (class in neurai.nn.neuron.snn_qif) SNNSRM (class in neurai.nn.neuron.snn_srm) SOFT (neurai.const.reset_type.ResetType attribute) Softmax (class in neurai.nn.layer.activate) Softmax2D (in module neurai.nn.layer.activate) softmax_cross_entropy() (in module neurai.nn.loss) Softmin (class in neurai.nn.layer.activate) Softplus (class in neurai.nn.layer.activate) SoftSign (class in neurai.nn.layer.activate) Solver (class in neurai.math.solver) SolverMethodODE (class in neurai.const.math) SolverMethodSDE (class in neurai.const.math) sparse_softmax_cross_entropy() (in module neurai.nn.loss) SparseLIF (class in neurai.nn.neuron.snn_lif) SpikeGenerator (class in neurai.nn.neuron.spike_generator) SpikeLoss (class in neurai.nn.loss) spikeTime() (neurai.nn.loss.SpikeLoss method) SRNN (class in neurai.nn.layer.recurrent) SRNNCellBase (class in neurai.nn.layer.recurrent) StaticLossScale (class in neurai.mp.scale) step() (neurai.math.ode.exp_euler.ExponentionalEuler method) (neurai.math.ode.runge_kutta.AdaptiveRungekutta method) (neurai.math.ode.runge_kutta.ExplictRungeKutta method) step_record() (neurai.monitor.MonitorBS method) StepMode (class in neurai.const.train) StepwiseNeuron (class in neurai.nn.neuron.neuron) SuperSpike (class in neurai.grads.surrogate_grad)

T

Tanh (class in neurai.nn.layer.activate) TdBatchNorm (class in neurai.nn.layer.normalization) TdLayer (class in neurai.nn.layer.normalization) trans() (neurai.manager.rng_manager.RngManager static method) transpose_output() (in module neurai.nn.layer.attention)

transpose_qkv() (in module neurai.nn.layer.attention) tree_flatten() (neurai.mp.scale.DynamicLossScale method) tree_unflatten() (neurai.mp.scale.DynamicLossScale class method) TuningConfig (class in neurai.finetune.finetune_config) TuningType (class in neurai.finetune.finetune_config)

U

UniformIniter (class in neurai.initializer.initializer) unscale() (neurai.mp.scale.DynamicLossScale method) (neurai.mp.scale.NoOpLossScale method) (neurai.mp.scale.StaticLossScale method)

UpSampleNearest (class in neurai.nn.layer.pooling)

V

Variable (class in neurai.manager.var_manager)

VarianceScalingIniter (class in neurai.initializer.initializer)

W

warn_if_not_floating() (in module neurai.mp.scale) weight_norm() (neurai.nn.layer.normalization.WeightNorm static method) WeightNorm (class in neurai.nn.layer.normalization)

with_output_dtype() (neurai.mp.policy.Policy method) write_file() (in module neurai.util.write) write_spike() (in module neurai.util.write) write_voltage() (in module neurai.util.write)

X

XavierUniformIniter (class in neurai.initializer.initializer)

Z

ZeroPad (class in neurai.nn.layer.padding)