LICENSE
MANIFEST.in
README.md
pyproject.toml
setup.py
LICENSES/NOTICE
ops/__init__.py
ops/_cuda.cuh
ops/autograd.py
ops/cuda_surrogate.cuh
ops/dispatch.py
ops/first_spike.py
ops/fp8_capability.py
ops/layout.py
ops/native_layout.cuh
ops/native_loader.py
ops/projection.py
ops/reset.py
ops/selection.py
ops/surrogate.py
ops/surrogate_dispatch.py
ops/surrogate_runtime.py
ops/triton_layout.py
ops/triton_missing.py
ops/triton_runtime.py
ops/triton_surrogate.py
ops/validation.py
ops/activation_aware_if/__init__.py
ops/activation_aware_if/cpu.py
ops/activation_aware_if/kernels.cuh
ops/activation_aware_if/native.cu
ops/activation_aware_if/native.py
ops/activation_aware_if/reference.py
ops/activation_aware_if/triton.py
ops/activation_aware_if/validation.py
ops/clif/__init__.py
ops/clif/autograd.py
ops/clif/cpu.py
ops/clif/reference.py
ops/clif/triton.py
ops/eif/__init__.py
ops/eif/autograd.py
ops/eif/cpu.py
ops/eif/kernels.cuh
ops/eif/native.cu
ops/eif/native.py
ops/eif/reference.py
ops/eif/triton.py
ops/flexsn/__init__.py
ops/flexsn/hop.py
ops/flexsn/info.py
ops/flexsn/kernel.py
ops/flexsn/launch.py
ops/flexsn/template.py
ops/flexsn/triton.py
ops/if_/__init__.py
ops/if_/autograd.py
ops/if_/cpu.py
ops/if_/kernels.cuh
ops/if_/native.cu
ops/if_/native.py
ops/if_/reference.py
ops/if_/triton.py
ops/if_/triton_precision.py
ops/if_linear/__init__.py
ops/if_linear/functional.py
ops/if_linear/kernels.cuh
ops/if_linear/native.cu
ops/ilif/__init__.py
ops/ilif/autograd.py
ops/ilif/cpu.py
ops/ilif/kernels.cuh
ops/ilif/native.cu
ops/ilif/native.py
ops/ilif/reference.py
ops/ilif/triton.py
ops/izhikevich/__init__.py
ops/izhikevich/autograd.py
ops/izhikevich/cpu.py
ops/izhikevich/kernels.cuh
ops/izhikevich/native.cu
ops/izhikevich/native.py
ops/izhikevich/reference.py
ops/izhikevich/triton.py
ops/lif/__init__.py
ops/lif/autograd.py
ops/lif/cpu.py
ops/lif/kernels.cuh
ops/lif/native.cu
ops/lif/native.py
ops/lif/reference.py
ops/lif/triton.py
ops/lif/triton_precision.py
ops/lif_linear/__init__.py
ops/lif_linear/functional.py
ops/lif_linear/kernels.cuh
ops/lif_linear/native.cu
ops/plif/__init__.py
ops/plif/autograd.py
ops/plif/cpu.py
ops/plif/kernels.cuh
ops/plif/native.cu
ops/plif/native.py
ops/plif/reference.py
ops/plif/triton.py
ops/plif/triton_precision.py
ops/qif/__init__.py
ops/qif/autograd.py
ops/qif/cpu.py
ops/qif/kernels.cuh
ops/qif/native.cu
ops/qif/native.py
ops/qif/reference.py
ops/qif/triton.py
ops/spike_compress/__init__.py
ops/spike_compress/cpu.py
ops/spike_compress/triton.py
ops/spike_compress/validation.py
ops/spike_linear/__init__.py
ops/spike_linear/kernels.cuh
ops/spike_linear/native.cu
ops/spike_linear/sparse.py
ops/stbif/__init__.py
ops/stbif/cpu.py
ops/stbif/kernels.cuh
ops/stbif/native.cu
ops/stbif/native.py
ops/stbif/reference.py
ops/stbif/triton.py
ops/stbif/validation.py
ops/tempotron/__init__.py
ops/tempotron/cpu.py
ops/tempotron/reference.py
ops/tempotron/triton.py
ops/tempotron/validation.py
ops/torch2triton/__init__.py
ops/torch2triton/graph2triton.py
ops/torch2triton/torch2graph.py
spikingjelly/__init__.py
spikingjelly/configure.py
spikingjelly/logger.py
spikingjelly/nsys.py
spikingjelly.egg-info/PKG-INFO
spikingjelly.egg-info/SOURCES.txt
spikingjelly.egg-info/dependency_links.txt
spikingjelly.egg-info/requires.txt
spikingjelly.egg-info/top_level.txt
spikingjelly/activation_based/__init__.py
spikingjelly/activation_based/_cuda_graph.py
spikingjelly/activation_based/_neuron.py
spikingjelly/activation_based/_neuron_layout.py
spikingjelly/activation_based/base.py
spikingjelly/activation_based/encoding.py
spikingjelly/activation_based/lava_exchange.py
spikingjelly/activation_based/lynxi_exchange.py
spikingjelly/activation_based/monitor.py
spikingjelly/activation_based/profiler.py
spikingjelly/activation_based/quantize.py
spikingjelly/activation_based/rnn.py
spikingjelly/activation_based/surrogate.py
spikingjelly/activation_based/ann2snn/__init__.py
spikingjelly/activation_based/ann2snn/converter.py
spikingjelly/activation_based/ann2snn/delay.py
spikingjelly/activation_based/ann2snn/modules.py
spikingjelly/activation_based/ann2snn/operators.py
spikingjelly/activation_based/ann2snn/qcfs.py
spikingjelly/activation_based/ann2snn/utils.py
spikingjelly/activation_based/ann2snn/examples/__init__.py
spikingjelly/activation_based/ann2snn/examples/bert_sst2_transformer_td_equivalent.py
spikingjelly/activation_based/ann2snn/examples/cnn_mnist.py
spikingjelly/activation_based/ann2snn/examples/imagenet_resnet18_ltb.py
spikingjelly/activation_based/ann2snn/examples/imagenet_vit_sta.py
spikingjelly/activation_based/ann2snn/examples/resnet18_cifar10.py
spikingjelly/activation_based/ann2snn/examples/roberta_spikezip_qann_synthetic.py
spikingjelly/activation_based/ann2snn/recipes/__init__.py
spikingjelly/activation_based/ann2snn/recipes/base.py
spikingjelly/activation_based/ann2snn/recipes/local_threshold_balancing.py
spikingjelly/activation_based/ann2snn/recipes/qwen2.py
spikingjelly/activation_based/ann2snn/recipes/rate_coding.py
spikingjelly/activation_based/ann2snn/recipes/spikezip_qann.py
spikingjelly/activation_based/ann2snn/recipes/sta_transformer.py
spikingjelly/activation_based/ann2snn/recipes/step_mode_adapters.py
spikingjelly/activation_based/ann2snn/recipes/transformer_td_equivalent.py
spikingjelly/activation_based/ann2snn/sample_models/__init__.py
spikingjelly/activation_based/ann2snn/sample_models/cifar10_resnet.py
spikingjelly/activation_based/ann2snn/sample_models/mnist_cnn.py
spikingjelly/activation_based/distributed/__init__.py
spikingjelly/activation_based/distributed/llm/__init__.py
spikingjelly/activation_based/distributed/llm/checkpoint.py
spikingjelly/activation_based/distributed/llm/config.py
spikingjelly/activation_based/distributed/llm/inference.py
spikingjelly/activation_based/distributed/llm/metrics.py
spikingjelly/activation_based/distributed/llm/planning.py
spikingjelly/activation_based/distributed/llm/sglang.py
spikingjelly/activation_based/distributed/llm/sglang_export.py
spikingjelly/activation_based/distributed/llm/temporal.py
spikingjelly/activation_based/distributed/llm/training.py
spikingjelly/activation_based/distributed/tensor_parallel/__init__.py
spikingjelly/activation_based/distributed/tensor_parallel/channel.py
spikingjelly/activation_based/distributed/vision/__init__.py
spikingjelly/activation_based/distributed/vision/config.py
spikingjelly/activation_based/distributed/vision/execution.py
spikingjelly/activation_based/distributed/vision/inference.py
spikingjelly/activation_based/distributed/vision/training.py
spikingjelly/activation_based/examples/__init__.py
spikingjelly/activation_based/examples/a2c.py
spikingjelly/activation_based/examples/cifar10_r11_enabling_spikebased_backpropagation.py
spikingjelly/activation_based/examples/classify_dvsg.py
spikingjelly/activation_based/examples/conv_fashion_mnist.py
spikingjelly/activation_based/examples/dqn_state.py
spikingjelly/activation_based/examples/lava_mnist.py
spikingjelly/activation_based/examples/lif_fc_mnist.py
spikingjelly/activation_based/examples/lynxi_fmnist_inference.py
spikingjelly/activation_based/examples/ppo.py
spikingjelly/activation_based/examples/rsnn_sequential_fmnist.py
spikingjelly/activation_based/examples/speech_commands.py
spikingjelly/activation_based/examples/spiking_a2c.py
spikingjelly/activation_based/examples/spiking_dqn_state.py
spikingjelly/activation_based/examples/spiking_lstm_sequential_mnist.py
spikingjelly/activation_based/examples/spiking_lstm_text.py
spikingjelly/activation_based/examples/spiking_ppo.py
spikingjelly/activation_based/examples/train_flexsn_compile.py
spikingjelly/activation_based/examples/train_imagenet.py
spikingjelly/activation_based/examples/common/__init__.py
spikingjelly/activation_based/examples/common/multiprocessing_env.py
spikingjelly/activation_based/examples/common/train_classify.py
spikingjelly/activation_based/examples/common/tv_ref_classify/__init__.py
spikingjelly/activation_based/examples/common/tv_ref_classify/presets.py
spikingjelly/activation_based/examples/common/tv_ref_classify/sampler.py
spikingjelly/activation_based/examples/common/tv_ref_classify/transforms.py
spikingjelly/activation_based/examples/common/tv_ref_classify/utils.py
spikingjelly/activation_based/examples/dsqn/train.py
spikingjelly/activation_based/examples/dsqn/ptan/__init__.py
spikingjelly/activation_based/examples/dsqn/ptan/actions.py
spikingjelly/activation_based/examples/dsqn/ptan/agent.py
spikingjelly/activation_based/examples/dsqn/ptan/experience.py
spikingjelly/activation_based/examples/dsqn/utils/__init__.py
spikingjelly/activation_based/examples/dsqn/utils/atari_wrappers.py
spikingjelly/activation_based/examples/dsqn/utils/common.py
spikingjelly/activation_based/examples/dsqn/utils/model.py
spikingjelly/activation_based/examples/ilc_san/core_cuda.py
spikingjelly/activation_based/examples/ilc_san/hybrid_td3_cuda_norm.py
spikingjelly/activation_based/examples/ilc_san/ilc_san.py
spikingjelly/activation_based/examples/ilc_san/replay_buffer_norm.py
spikingjelly/activation_based/examples/ilc_san/test_hybrid_td3_cpu.py
spikingjelly/activation_based/examples/memopt/data_module.py
spikingjelly/activation_based/examples/memopt/lightning_callbacks.py
spikingjelly/activation_based/examples/memopt/lightning_modules.py
spikingjelly/activation_based/examples/memopt/loss.py
spikingjelly/activation_based/examples/memopt/models.py
spikingjelly/activation_based/examples/memopt/train.py
spikingjelly/activation_based/examples/noisy_san/core_cuda.py
spikingjelly/activation_based/examples/noisy_san/hybrid_td3_cuda_norm.py
spikingjelly/activation_based/examples/noisy_san/noisy_san.py
spikingjelly/activation_based/examples/noisy_san/replay_buffer_norm.py
spikingjelly/activation_based/examples/noisy_san/test_hybrid_td3_cpu.py
spikingjelly/activation_based/examples/plasticity/__init__.py
spikingjelly/activation_based/examples/plasticity/mstdp.py
spikingjelly/activation_based/examples/plasticity/mstdpet.py
spikingjelly/activation_based/examples/plasticity/stdp_mnist.py
spikingjelly/activation_based/examples/plasticity/stdp_trace.py
spikingjelly/activation_based/examples/profiling/nsys_lif.py
spikingjelly/activation_based/examples/profiling/nsys_multigpu.py
spikingjelly/activation_based/functional/__init__.py
spikingjelly/activation_based/functional/conv_bn_fusion.py
spikingjelly/activation_based/functional/forward.py
spikingjelly/activation_based/functional/layer.py
spikingjelly/activation_based/functional/learning.py
spikingjelly/activation_based/functional/loss.py
spikingjelly/activation_based/functional/misc.py
spikingjelly/activation_based/functional/net_config.py
spikingjelly/activation_based/functional/neuron.py
spikingjelly/activation_based/functional/spike.py
spikingjelly/activation_based/layer/__init__.py
spikingjelly/activation_based/layer/attention.py
spikingjelly/activation_based/layer/bn.py
spikingjelly/activation_based/layer/container.py
spikingjelly/activation_based/layer/dropout.py
spikingjelly/activation_based/layer/misc.py
spikingjelly/activation_based/layer/online_learning.py
spikingjelly/activation_based/layer/stateless_wrapper.py
spikingjelly/activation_based/learning/__init__.py
spikingjelly/activation_based/learning/online_learning.py
spikingjelly/activation_based/learning/plasticity.py
spikingjelly/activation_based/memopt/__init__.py
spikingjelly/activation_based/memopt/checkpointing.py
spikingjelly/activation_based/memopt/compress.py
spikingjelly/activation_based/memopt/pipeline.py
spikingjelly/activation_based/model/__init__.py
spikingjelly/activation_based/model/masnn.py
spikingjelly/activation_based/model/maxformer.py
spikingjelly/activation_based/model/ms_resnet.py
spikingjelly/activation_based/model/parametric_lif_net.py
spikingjelly/activation_based/model/qkformer.py
spikingjelly/activation_based/model/sew_resnet.py
spikingjelly/activation_based/model/snas_net.py
spikingjelly/activation_based/model/spike_dhs.py
spikingjelly/activation_based/model/spike_driven_transformer.py
spikingjelly/activation_based/model/spikformer.py
spikingjelly/activation_based/model/spiking_resnet.py
spikingjelly/activation_based/model/spiking_vgg.py
spikingjelly/activation_based/model/spiking_vggws_ottt.py
spikingjelly/activation_based/neuron/__init__.py
spikingjelly/activation_based/neuron/adapt.py
spikingjelly/activation_based/neuron/base_node.py
spikingjelly/activation_based/neuron/dsr.py
spikingjelly/activation_based/neuron/few_spike.py
spikingjelly/activation_based/neuron/flexsn.py
spikingjelly/activation_based/neuron/flexsn_hop.py
spikingjelly/activation_based/neuron/flexsn_trace.py
spikingjelly/activation_based/neuron/ilif.py
spikingjelly/activation_based/neuron/integrate_and_fire.py
spikingjelly/activation_based/neuron/inter_layer_connection.py
spikingjelly/activation_based/neuron/lif.py
spikingjelly/activation_based/neuron/lif_variants.py
spikingjelly/activation_based/neuron/mpbn.py
spikingjelly/activation_based/neuron/noisy.py
spikingjelly/activation_based/neuron/nonlinear_if.py
spikingjelly/activation_based/neuron/online_learning.py
spikingjelly/activation_based/neuron/plif.py
spikingjelly/activation_based/neuron/psn.py
spikingjelly/activation_based/neuron/resonate.py
spikingjelly/activation_based/neuron/spikezip.py
spikingjelly/activation_based/neuron/ternary.py
spikingjelly/activation_based/nir_exchange/__init__.py
spikingjelly/activation_based/nir_exchange/from_nir.py
spikingjelly/activation_based/nir_exchange/to_nir.py
spikingjelly/activation_based/op_counter/__init__.py
spikingjelly/activation_based/op_counter/_sparse_memory.py
spikingjelly/activation_based/op_counter/ac.py
spikingjelly/activation_based/op_counter/analytical_energy.py
spikingjelly/activation_based/op_counter/base.py
spikingjelly/activation_based/op_counter/flop.py
spikingjelly/activation_based/op_counter/mac.py
spikingjelly/activation_based/op_counter/memory_access.py
spikingjelly/activation_based/op_counter/neuromorphic_memory_access.py
spikingjelly/activation_based/op_counter/neuron_state.py
spikingjelly/activation_based/op_counter/simple_energy.py
spikingjelly/activation_based/op_counter/synop.py
spikingjelly/activation_based/op_counter/neuromc/__init__.py
spikingjelly/activation_based/op_counter/neuromc/config.py
spikingjelly/activation_based/op_counter/neuromc/core.py
spikingjelly/activation_based/op_counter/neuromc/mapping.py
spikingjelly/activation_based/op_counter/neuromc/utils.py
spikingjelly/activation_based/op_counter/spikesim/__init__.py
spikingjelly/activation_based/op_counter/spikesim/config.py
spikingjelly/activation_based/op_counter/spikesim/core.py
spikingjelly/activation_based/op_counter/spikesim/counter.py
spikingjelly/activation_based/op_counter/spikesim/formulas.py
spikingjelly/activation_based/precision/__init__.py
spikingjelly/activation_based/precision/api.py
spikingjelly/activation_based/precision/capability.py
spikingjelly/activation_based/precision/config.py
spikingjelly/activation_based/precision/convert.py
spikingjelly/activation_based/precision/float8_attention.py
spikingjelly/activation_based/precision/float8_base.py
spikingjelly/activation_based/precision/float8_conv.py
spikingjelly/activation_based/precision/float8_te.py
spikingjelly/activation_based/precision/policy.py
spikingjelly/datasets/__init__.py
spikingjelly/datasets/asl_dvs.py
spikingjelly/datasets/base.py
spikingjelly/datasets/bullying10k.py
spikingjelly/datasets/cifar10_dvs.py
spikingjelly/datasets/dvs128_gesture.py
spikingjelly/datasets/dvs_lip.py
spikingjelly/datasets/es_imagenet.py
spikingjelly/datasets/hardvs.py
spikingjelly/datasets/n_caltech101.py
spikingjelly/datasets/n_mnist.py
spikingjelly/datasets/nav_gesture.py
spikingjelly/datasets/shd.py
spikingjelly/datasets/speechcommands.py
spikingjelly/datasets/transform.py
spikingjelly/datasets/utils.py
spikingjelly/timing_based/__init__.py
spikingjelly/timing_based/encoding.py
spikingjelly/timing_based/functional.py
spikingjelly/timing_based/neuron.py
spikingjelly/timing_based/solver.py
spikingjelly/timing_based/examples/__init__.py
spikingjelly/timing_based/examples/cubic_conv_mnist.py
spikingjelly/timing_based/examples/cubic_srm_mnist.py
spikingjelly/timing_based/examples/srm_mnist.py
spikingjelly/timing_based/examples/tempotron_mnist.py
spikingjelly/visualizing/__init__.py
spikingjelly/visualizing/_utils.py
spikingjelly/visualizing/bar3d.py
spikingjelly/visualizing/feature_map.py
spikingjelly/visualizing/heatmap.py
spikingjelly/visualizing/spikes.py
test/test_configure.py
test/test_logger.py
test/test_visualizing.py