LICENSE
MANIFEST.in
README.md
pyproject.toml
docs/API/agent.en.md
docs/API/llm_search_agent.en.md
docs/API/load_data.en.md
docs/API/model.en.md
docs/API/model.params.en.md
docs/API/model.test.en.md
docs/API/model.train.en.md
docs/API/summary.en.md
docs/API/toChip.en.md
docs/API/visualize.en.md
docs/Demo/chip_deploy.en.md
docs/Demo/complex_scenarios.en.md
docs/Demo/demo.en.md
docs/General/overview.en.md
src/bcijelly/__init__.py
src/bcijelly/llm_search_agent.py
src/bcijelly/to_chip.py
src/bcijelly/visualize.py
src/bcijelly.egg-info/PKG-INFO
src/bcijelly.egg-info/SOURCES.txt
src/bcijelly.egg-info/dependency_links.txt
src/bcijelly.egg-info/requires.txt
src/bcijelly.egg-info/top_level.txt
src/bcijelly/agent/__init__.py
src/bcijelly/agent/api.py
src/bcijelly/agent/data_utils.py
src/bcijelly/agent/evaluator.py
src/bcijelly/agent/model_builder.py
src/bcijelly/agent/registry.py
src/bcijelly/agent/searcher.py
src/bcijelly/agent/trainer.py
src/bcijelly/agent/utils.py
src/bcijelly/agent/model/__init__.py
src/bcijelly/agent/model/attention/SelfAttention.py
src/bcijelly/agent/model/backbone/resnet.py
src/bcijelly/agent/model/conv/DepthwiseSeparableConvolution.py
src/bcijelly/agent/model/mlp/resmlp.py
src/bcijelly/agent/model/rep/repvgg.py
src/bcijelly/dataset/_discovery.py
src/bcijelly/dataset/_io.py
src/bcijelly/dataset/_load_data_capability.py
src/bcijelly/dataset/_loading.py
src/bcijelly/dataset/_types.py
src/bcijelly/dataset/load_data.py
src/bcijelly/llm/__init__.py
src/bcijelly/llm/_config.py
src/bcijelly/llm/_types.py
src/bcijelly/llm/catalog.py
src/bcijelly/llm/constants.py
src/bcijelly/llm/datasets_user.py
src/bcijelly/llm/encoder_baseline.py
src/bcijelly/llm/evaluator.py
src/bcijelly/llm/foundation_encoder.py
src/bcijelly/llm/llm_advisor.py
src/bcijelly/llm/llm_module_designer.py
src/bcijelly/llm/module_index.py
src/bcijelly/llm/native_pipeline.py
src/bcijelly/llm/pipeline.py
src/bcijelly/llm/report_utils.py
src/bcijelly/llm/session_encoder.py
src/bcijelly/llm/split_data.py
src/bcijelly/llm/module/__init__.py
src/bcijelly/llm/module/attention/A2Atttention.py
src/bcijelly/llm/module/attention/ACmixAttention.py
src/bcijelly/llm/module/attention/AFT.py
src/bcijelly/llm/module/attention/Axial_attention.py
src/bcijelly/llm/module/attention/BAM.py
src/bcijelly/llm/module/attention/CBAM.py
src/bcijelly/llm/module/attention/CoAtNet.py
src/bcijelly/llm/module/attention/CoTAttention.py
src/bcijelly/llm/module/attention/CoordAttention.py
src/bcijelly/llm/module/attention/CrissCrossAttention.py
src/bcijelly/llm/module/attention/Crossformer.py
src/bcijelly/llm/module/attention/DANet.py
src/bcijelly/llm/module/attention/DAT.py
src/bcijelly/llm/module/attention/ECAAttention.py
src/bcijelly/llm/module/attention/EMSA.py
src/bcijelly/llm/module/attention/ExternalAttention.py
src/bcijelly/llm/module/attention/HaloAttention.py
src/bcijelly/llm/module/attention/MOATransformer.py
src/bcijelly/llm/module/attention/MUSEAttention.py
src/bcijelly/llm/module/attention/MobileViTAttention.py
src/bcijelly/llm/module/attention/MobileViTv2Attention.py
src/bcijelly/llm/module/attention/OutlookAttention.py
src/bcijelly/llm/module/attention/PSA.py
src/bcijelly/llm/module/attention/ParNetAttention.py
src/bcijelly/llm/module/attention/PolarizedSelfAttention.py
src/bcijelly/llm/module/attention/ResidualAttention.py
src/bcijelly/llm/module/attention/S2Attention.py
src/bcijelly/llm/module/attention/SEAttention.py
src/bcijelly/llm/module/attention/SGE.py
src/bcijelly/llm/module/attention/SKAttention.py
src/bcijelly/llm/module/attention/SelfAttention.py
src/bcijelly/llm/module/attention/ShuffleAttention.py
src/bcijelly/llm/module/attention/SimAM.py
src/bcijelly/llm/module/attention/SimplifiedSelfAttention.py
src/bcijelly/llm/module/attention/TripletAttention.py
src/bcijelly/llm/module/attention/UFOAttention.py
src/bcijelly/llm/module/attention/ViP.py
src/bcijelly/llm/module/attention/gfnet.py
src/bcijelly/llm/module/backbone/CMT.py
src/bcijelly/llm/module/backbone/CPVT.py
src/bcijelly/llm/module/backbone/CaiT.py
src/bcijelly/llm/module/backbone/CeiT.py
src/bcijelly/llm/module/backbone/CoaT.py
src/bcijelly/llm/module/backbone/ConTNet.py
src/bcijelly/llm/module/backbone/ConViT.py
src/bcijelly/llm/module/backbone/Container.py
src/bcijelly/llm/module/backbone/ConvMixer.py
src/bcijelly/llm/module/backbone/CrossViT.py
src/bcijelly/llm/module/backbone/DViT.py
src/bcijelly/llm/module/backbone/DeiT.py
src/bcijelly/llm/module/backbone/EfficientFormer.py
src/bcijelly/llm/module/backbone/HATNet.py
src/bcijelly/llm/module/backbone/LeViT.py
src/bcijelly/llm/module/backbone/MobileNetV3.py
src/bcijelly/llm/module/backbone/MobileViT.py
src/bcijelly/llm/module/backbone/PIT.py
src/bcijelly/llm/module/backbone/PVT.py
src/bcijelly/llm/module/backbone/PatchConvnet.py
src/bcijelly/llm/module/backbone/ShuffleTransformer.py
src/bcijelly/llm/module/backbone/TnT.py
src/bcijelly/llm/module/backbone/VOLO.py
src/bcijelly/llm/module/backbone/convnextv2.py
src/bcijelly/llm/module/backbone/resnet.py
src/bcijelly/llm/module/backbone/resnext.py
src/bcijelly/llm/module/backbone/swin_transformer.py
src/bcijelly/llm/module/backbone/swin_transformer_v2.py
src/bcijelly/llm/module/backbone/swin_transformer_v2_cr.py
src/bcijelly/llm/module/conv/CondConv.py
src/bcijelly/llm/module/conv/DepthwiseSeparableConvolution.py
src/bcijelly/llm/module/conv/DynamicConv.py
src/bcijelly/llm/module/conv/HorNet.py
src/bcijelly/llm/module/conv/Involution.py
src/bcijelly/llm/module/conv/MBConv.py
src/bcijelly/llm/module/mlp/g_mlp.py
src/bcijelly/llm/module/mlp/mlp_mixer.py
src/bcijelly/llm/module/mlp/repmlp.py
src/bcijelly/llm/module/mlp/resmlp.py
src/bcijelly/llm/module/mlp/sMLP_block.py
src/bcijelly/llm/module/mlp/vip-mlp.py
src/bcijelly/llm/module/rep/acnet.py
src/bcijelly/llm/module/rep/ddb.py
src/bcijelly/llm/module/rep/mobileone.py
src/bcijelly/llm/module/rep/repvgg.py
src/bcijelly/llm/module_library/__init__.py
src/bcijelly/llm/module_library/attention.py
src/bcijelly/llm/module_library/backbone.py
src/bcijelly/llm/module_library/conv.py
src/bcijelly/llm/module_library/fc.py
src/bcijelly/llm/module_library/designed/__init__.py
src/bcijelly/llm/module_library/designed/acmix_temporal_attention_t.py
src/bcijelly/llm/module_library/designed/adaptive_residual_ffn.py
src/bcijelly/llm/module_library/designed/adaptive_temporal_sampling_conv.py
src/bcijelly/llm/module_library/designed/aft_temporal_pool_t.py
src/bcijelly/llm/module_library/designed/asym_conv1d_acblock_t.py
src/bcijelly/llm/module_library/designed/axial_temporal_attention_t.py
src/bcijelly/llm/module_library/designed/bicoherence_phase_coupling_film_conv.py
src/bcijelly/llm/module_library/designed/bicoherence_qpc_film_conv.py
src/bcijelly/llm/module_library/designed/bicoherence_quadratic_phase_coupling_conv.py
src/bcijelly/llm/module_library/designed/bicoherence_quadratic_phase_coupling_film_conv.py
src/bcijelly/llm/module_library/designed/bispectral_phase_coupling_conv.py
src/bcijelly/llm/module_library/designed/bispectral_quadratic_coupling_conv.py
src/bcijelly/llm/module_library/designed/cait_layerscale_temporal_conv_t.py
src/bcijelly/llm/module_library/designed/causal_statespace_oscillator_conv.py
src/bcijelly/llm/module_library/designed/cbam_temporal_t.py
src/bcijelly/llm/module_library/designed/channel_attention_conv_pool.py
src/bcijelly/llm/module_library/designed/channel_covariance_pool.py
src/bcijelly/llm/module_library/designed/channel_wise_temporal_attn_pool.py
src/bcijelly/llm/module_library/designed/cmt_inverted_residual_temporal_t.py
src/bcijelly/llm/module_library/designed/coactivation_graph_diffusion_conv.py
src/bcijelly/llm/module_library/designed/cofluctuation_edge_event_conv.py
src/bcijelly/llm/module_library/designed/common_mode_rejection_conv.py
src/bcijelly/llm/module_library/designed/cond_conv1d_temporal_t.py
src/bcijelly/llm/module_library/designed/container_temporal_mixer_t.py
src/bcijelly/llm/module_library/designed/conv_mixer_temporal_t.py
src/bcijelly/llm/module_library/designed/conv_rel_pos_enc_1d_t.py
src/bcijelly/llm/module_library/designed/conv_self_attention_pool.py
src/bcijelly/llm/module_library/designed/convnextv2_temporal_block_t.py
src/bcijelly/llm/module_library/designed/cosine_prototype_residual_ffn.py
src/bcijelly/llm/module_library/designed/cot_temporal_attention_t.py
src/bcijelly/llm/module_library/designed/criss_cross_temporal_attention_t.py
src/bcijelly/llm/module_library/designed/cross_channel_correlation_pool.py
src/bcijelly/llm/module_library/designed/cross_channel_leadlag_synchrony_conv.py
src/bcijelly/llm/module_library/designed/cross_channel_phase_locking_conv.py
src/bcijelly/llm/module_library/designed/cross_frequency_coupling_pool.py
src/bcijelly/llm/module_library/designed/cross_scale_temporal_fusion_t.py
src/bcijelly/llm/module_library/designed/cross_spectral_coherence_pool.py
src/bcijelly/llm/module_library/designed/crossformer_temporal_attn_pool_t.py
src/bcijelly/llm/module_library/designed/deit_dual_token_temporal_conv_t.py
src/bcijelly/llm/module_library/designed/depthwise_separable_temporal_conv_t.py
src/bcijelly/llm/module_library/designed/detrended_fluctuation_film_conv.py
src/bcijelly/llm/module_library/designed/detrended_fluctuation_scaling_conv.py
src/bcijelly/llm/module_library/designed/deviation_prior_attention_pool.py
src/bcijelly/llm/module_library/designed/diagonal_state_space_pool.py
src/bcijelly/llm/module_library/designed/difference_of_exponentials_conv.py
src/bcijelly/llm/module_library/designed/diverse_branch_temporal_block_t.py
src/bcijelly/llm/module_library/designed/divisive_norm_temporal_conv.py
src/bcijelly/llm/module_library/designed/divnorm_gain_control_ffn.py
src/bcijelly/llm/module_library/designed/double_attention_temporal_pool_t.py
src/bcijelly/llm/module_library/designed/doubly_stochastic_sinkhorn_conv.py
src/bcijelly/llm/module_library/designed/dual_attention_temporal_t.py
src/bcijelly/llm/module_library/designed/dynamic_op_mixture_ffn.py
src/bcijelly/llm/module_library/designed/dynamic_temporal_conv_t.py
src/bcijelly/llm/module_library/designed/eca_temporal_channel_attention_t.py
src/bcijelly/llm/module_library/designed/efficientformer_temporal_metablock_t.py
src/bcijelly/llm/module_library/designed/external_attention_temporal_t.py
src/bcijelly/llm/module_library/designed/factorized_bilinear_residual_ffn.py
src/bcijelly/llm/module_library/designed/fourier_kan_activation_ffn.py
src/bcijelly/llm/module_library/designed/fourier_phase_group_delay_conv.py
src/bcijelly/llm/module_library/designed/gain_invariant_descriptor_panel_conv.py
src/bcijelly/llm/module_library/designed/gated_dilated_temporal_pool.py
src/bcijelly/llm/module_library/designed/gated_positional_temporal_conv_t.py
src/bcijelly/llm/module_library/designed/gated_recurrent_dynamics_pool.py
src/bcijelly/llm/module_library/designed/gated_temporal_depthwise_conv.py
src/bcijelly/llm/module_library/designed/gated_temporal_glu_conv.py
src/bcijelly/llm/module_library/designed/gaussian_mixture_attention_pool.py
src/bcijelly/llm/module_library/designed/gaussian_temporal_field_pool.py
src/bcijelly/llm/module_library/designed/grouped_attention_covariance_pool.py
src/bcijelly/llm/module_library/designed/halo_local_temporal_attention_t.py
src/bcijelly/llm/module_library/designed/higuchi_fractal_dimension_film_conv.py
src/bcijelly/llm/module_library/designed/hilbert_instantaneous_frequency_conv.py
src/bcijelly/llm/module_library/designed/hilbert_instantaneous_frequency_film_conv.py
src/bcijelly/llm/module_library/designed/hjorth_dynamics_conv.py
src/bcijelly/llm/module_library/designed/input_dependent_gating_ffn.py
src/bcijelly/llm/module_library/designed/intrinsic_timescale_acf_conv.py
src/bcijelly/llm/module_library/designed/inverted_residual_temporal_t.py
src/bcijelly/llm/module_library/designed/lag_autocorrelation_temporal_conv.py
src/bcijelly/llm/module_library/designed/lagged_cross_correlation_pool.py
src/bcijelly/llm/module_library/designed/layerwise_gain_modulated_ffn.py
src/bcijelly/llm/module_library/designed/leaky_integrator_temporal_conv.py
src/bcijelly/llm/module_library/designed/leff_temporal_1d_t.py
src/bcijelly/llm/module_library/designed/legendre_temporal_moment_pool.py
src/bcijelly/llm/module_library/designed/level_crossing_rate_conv.py
src/bcijelly/llm/module_library/designed/levit_temporal_mixer_t.py
src/bcijelly/llm/module_library/designed/local_matched_template_ncc_conv.py
src/bcijelly/llm/module_library/designed/local_population_synchrony_conv.py
src/bcijelly/llm/module_library/designed/local_standardized_moment_conv.py
src/bcijelly/llm/module_library/designed/low_frequency_trajectory_pool.py
src/bcijelly/llm/module_library/designed/low_rank_volterra_temporal_pool.py
src/bcijelly/llm/module_library/designed/matrix_sqrt_correlation_pool.py
src/bcijelly/llm/module_library/designed/moa_temporal_pool_attention_t.py
src/bcijelly/llm/module_library/designed/mobilenetv3_temporal_block_t.py
src/bcijelly/llm/module_library/designed/mobileone_temporal_block_t.py
src/bcijelly/llm/module_library/designed/mobilevit_temporal_block_t.py
src/bcijelly/llm/module_library/designed/mobilevitv2_temporal_attention_t.py
src/bcijelly/llm/module_library/designed/moe_residual_ffn_zeroinit.py
src/bcijelly/llm/module_library/designed/morlet_oscillatory_conv.py
src/bcijelly/llm/module_library/designed/morphological_temporal_conv.py
src/bcijelly/llm/module_library/designed/multi_query_attn_pool.py
src/bcijelly/llm/module_library/designed/multi_query_temporal_attention_pool.py
src/bcijelly/llm/module_library/designed/multi_timescale_leaky_integrator_pool.py
src/bcijelly/llm/module_library/designed/multifractal_spectrum_width_film_conv.py
src/bcijelly/llm/module_library/designed/multihead_attn_pool_residual.py
src/bcijelly/llm/module_library/designed/multiscale_dilated_conv.py
src/bcijelly/llm/module_library/designed/multiscale_gated_depthwise_conv.py
src/bcijelly/llm/module_library/designed/multiscale_recency_integrator_pool.py
src/bcijelly/llm/module_library/designed/multiscale_sample_entropy_conv.py
src/bcijelly/llm/module_library/designed/multitaper_spectral_film_conv.py
src/bcijelly/llm/module_library/designed/muse_temporal_conv_t.py
src/bcijelly/llm/module_library/designed/netvlad_temporal_pool.py
src/bcijelly/llm/module_library/designed/normalized_cross_correlation_conv.py
src/bcijelly/llm/module_library/designed/onset_gated_attention_pool.py
src/bcijelly/llm/module_library/designed/ordinal_pattern_symbolic_conv.py
src/bcijelly/llm/module_library/designed/parnet_temporal_attention_t.py
src/bcijelly/llm/module_library/designed/patchconvnet_temporal_block_t.py
src/bcijelly/llm/module_library/designed/phase_amplitude_coupling_conv.py
src/bcijelly/llm/module_library/designed/phase_amplitude_coupling_film_conv.py
src/bcijelly/llm/module_library/designed/phase_locking_value_film_conv.py
src/bcijelly/llm/module_library/designed/phase_portrait_angular_velocity_conv.py
src/bcijelly/llm/module_library/designed/pid_control_pool.py
src/bcijelly/llm/module_library/designed/piecewise_linear_ensemble_ffn.py
src/bcijelly/llm/module_library/designed/pit_temporal_pool_t.py
src/bcijelly/llm/module_library/designed/polarized_self_attention_1d_t.py
src/bcijelly/llm/module_library/designed/population_activity_histogram_conv.py
src/bcijelly/llm/module_library/designed/population_dimensionality_geometry_film_conv.py
src/bcijelly/llm/module_library/designed/population_effective_dimensionality_conv.py
src/bcijelly/llm/module_library/designed/population_lagged_connectivity_conv.py
src/bcijelly/llm/module_library/designed/population_latency_sequence_conv.py
src/bcijelly/llm/module_library/designed/population_lead_lag_directionality_film_conv.py
src/bcijelly/llm/module_library/designed/population_rank_code_conv.py
src/bcijelly/llm/module_library/designed/population_recurrence_conv.py
src/bcijelly/llm/module_library/designed/population_state_codebook_conv.py
src/bcijelly/llm/module_library/designed/population_trajectory_kinematics_conv.py
src/bcijelly/llm/module_library/designed/population_trajectory_tangling_conv.py
src/bcijelly/llm/module_library/designed/population_whitening_conv.py
src/bcijelly/llm/module_library/designed/prototype_rbf_residual_ffn.py
src/bcijelly/llm/module_library/designed/quadratic_interaction_bottleneck_ffn.py
src/bcijelly/llm/module_library/designed/rank_quantile_pool.py
src/bcijelly/llm/module_library/designed/recency_biased_temporal_attention_pool.py
src/bcijelly/llm/module_library/designed/recurrence_determinism_film_conv.py
src/bcijelly/llm/module_library/designed/recurrence_quantification_film_conv.py
src/bcijelly/llm/module_library/designed/repmlp_temporal_t.py
src/bcijelly/llm/module_library/designed/res_mlp_temporal_mixer_t.py
src/bcijelly/llm/module_library/designed/residual_attention_temporal_pool_t.py
src/bcijelly/llm/module_library/designed/residual_gated_ffn_zeroinit.py
src/bcijelly/llm/module_library/designed/resnext_temporal_block_t.py
src/bcijelly/llm/module_library/designed/se_temporal_recalibration_t.py
src/bcijelly/llm/module_library/designed/selective_kernel_temporal_conv.py
src/bcijelly/llm/module_library/designed/selective_state_integrator_pool.py
src/bcijelly/llm/module_library/designed/set_transformer_pool.py
src/bcijelly/llm/module_library/designed/shuffle_temporal_attention_t.py
src/bcijelly/llm/module_library/designed/shuffle_window_temporal_conv_t.py
src/bcijelly/llm/module_library/designed/simam_temporal_pool_t.py
src/bcijelly/llm/module_library/designed/simplified_self_attention_temporal_pool_t.py
src/bcijelly/llm/module_library/designed/sk_temporal_select_t.py
src/bcijelly/llm/module_library/designed/smlp_temporal_block_t.py
src/bcijelly/llm/module_library/designed/soft_dtw_template_conv.py
src/bcijelly/llm/module_library/designed/sparse_topk_residual_ffn.py
src/bcijelly/llm/module_library/designed/spectral_band_pool.py
src/bcijelly/llm/module_library/designed/spectral_entropy_film_conv.py
src/bcijelly/llm/module_library/designed/spectral_global_filter_conv.py
src/bcijelly/llm/module_library/designed/spectral_logpower_envelope_conv.py
src/bcijelly/llm/module_library/designed/spectral_quantile_frequency_conv.py
src/bcijelly/llm/module_library/designed/spectral_wiener_filter_pool.py
src/bcijelly/llm/module_library/designed/structure_function_scaling_film_conv.py
src/bcijelly/llm/module_library/designed/swin_temporal_conv_t.py
src/bcijelly/llm/module_library/designed/swinv2_temporal_local_block_t.py
src/bcijelly/llm/module_library/designed/tapped_trajectory_decoder_pool.py
src/bcijelly/llm/module_library/designed/temporal_adaptive_ffn.py
src/bcijelly/llm/module_library/designed/temporal_bam_t.py
src/bcijelly/llm/module_library/designed/temporal_bottleneck_resnet_t.py
src/bcijelly/llm/module_library/designed/temporal_cepstral_envelope_conv.py
src/bcijelly/llm/module_library/designed/temporal_coatnet_t.py
src/bcijelly/llm/module_library/designed/temporal_conditional_pos_encoding_t.py
src/bcijelly/llm/module_library/designed/temporal_cont_block_t.py
src/bcijelly/llm/module_library/designed/temporal_conv_lstm_cell.py
src/bcijelly/llm/module_library/designed/temporal_convolutional_attention_pool.py
src/bcijelly/llm/module_library/designed/temporal_coord_attention_t.py
src/bcijelly/llm/module_library/designed/temporal_copula_rank_conv.py
src/bcijelly/llm/module_library/designed/temporal_cross_attention_covariance_pool.py
src/bcijelly/llm/module_library/designed/temporal_deformable_attention_t.py
src/bcijelly/llm/module_library/designed/temporal_derivative_conv.py
src/bcijelly/llm/module_library/designed/temporal_emsa_t.py
src/bcijelly/llm/module_library/designed/temporal_global_filter_t.py
src/bcijelly/llm/module_library/designed/temporal_gmlp_gating_t.py
src/bcijelly/llm/module_library/designed/temporal_group_enhance_t.py
src/bcijelly/llm/module_library/designed/temporal_hornet_gnconv_t.py
src/bcijelly/llm/module_library/designed/temporal_involution_t.py
src/bcijelly/llm/module_library/designed/temporal_latency_code_conv.py
src/bcijelly/llm/module_library/designed/temporal_mbconv_t.py
src/bcijelly/llm/module_library/designed/temporal_mixer_block_t.py
src/bcijelly/llm/module_library/designed/temporal_mobilevit_attention_t.py
src/bcijelly/llm/module_library/designed/temporal_outlook_attention_t.py
src/bcijelly/llm/module_library/designed/temporal_outlook_attention_t2.py
src/bcijelly/llm/module_library/designed/temporal_pyramid_attention_pool.py
src/bcijelly/llm/module_library/designed/temporal_pyramid_squeeze_attention_t.py
src/bcijelly/llm/module_library/designed/temporal_reattention_t.py
src/bcijelly/llm/module_library/designed/temporal_relational_graph_pool.py
src/bcijelly/llm/module_library/designed/temporal_repvgg_block_t.py
src/bcijelly/llm/module_library/designed/temporal_s2_shift_attention_t.py
src/bcijelly/llm/module_library/designed/temporal_se_conv.py
src/bcijelly/llm/module_library/designed/temporal_self_attention_pool_t.py
src/bcijelly/llm/module_library/designed/temporal_spectral_conv.py
src/bcijelly/llm/module_library/designed/temporal_sra_block_t.py
src/bcijelly/llm/module_library/designed/temporal_statistics_pool.py
src/bcijelly/llm/module_library/designed/temporal_swin_v2_window_attn_t.py
src/bcijelly/llm/module_library/designed/temporal_triplet_attention_t.py
src/bcijelly/llm/module_library/designed/time_irreversibility_conv.py
src/bcijelly/llm/module_library/designed/tnt_temporal_block_t.py
src/bcijelly/llm/module_library/designed/ufo_temporal_attention_pool_t.py
src/bcijelly/llm/module_library/designed/wavelet_scattering_conv.py
src/bcijelly/llm/module_library/designed/weighted_permute_temporal_mlp_t.py
src/bcijelly/llm/module_library/designed/weighted_permute_temporal_t.py
src/bcijelly/llm/module_library/designed/wiener_laguerre_pool.py
src/bcijelly/model/__init__.py
src/bcijelly/model/_factory.py
src/bcijelly/model/_train_utils.py
src/bcijelly/model/presets.py
src/bcijelly/model/_cycle_gan/__init__.py
src/bcijelly/model/_cycle_gan/classifier.py
src/bcijelly/model/_cycle_gan/plot_csv.py
src/bcijelly/model/_cycle_gan/regressor.py
src/bcijelly/model/_cycle_gan/shared_utils.py
src/bcijelly/model/_dfine/__init__.py
src/bcijelly/model/_dfine/model.py
src/bcijelly/model/_dfine/regressor.py
src/bcijelly/model/_dfine/shared_utils.py
src/bcijelly/model/_fenet/__init__.py
src/bcijelly/model/_fenet/model.py
src/bcijelly/model/_fenet/regressor.py
src/bcijelly/model/_fenet/shared_utils.py
src/bcijelly/model/_kf/__init__.py
src/bcijelly/model/_kf/regressor.py
src/bcijelly/model/_lfads/__init__.py
src/bcijelly/model/_lfads/classifier.py
src/bcijelly/model/_lfads/regressor.py
src/bcijelly/model/_lfads/shared_utils.py
src/bcijelly/model/_lfads/lfads_torch/__init__.py
src/bcijelly/model/_lfads/lfads_torch/callbacks.py
src/bcijelly/model/_lfads/lfads_torch/datamodules.py
src/bcijelly/model/_lfads/lfads_torch/metrics.py
src/bcijelly/model/_lfads/lfads_torch/model.py
src/bcijelly/model/_lfads/lfads_torch/run_model.py
src/bcijelly/model/_lfads/lfads_torch/tuples.py
src/bcijelly/model/_lfads/lfads_torch/utils.py
src/bcijelly/model/_lfads/lfads_torch/utils_autoconfig.py
src/bcijelly/model/_lfads/lfads_torch/extensions/__init__.py
src/bcijelly/model/_lfads/lfads_torch/extensions/nlb.py
src/bcijelly/model/_lfads/lfads_torch/extensions/tune.py
src/bcijelly/model/_lfads/lfads_torch/modules/__init__.py
src/bcijelly/model/_lfads/lfads_torch/modules/augmentations.py
src/bcijelly/model/_lfads/lfads_torch/modules/decoder.py
src/bcijelly/model/_lfads/lfads_torch/modules/encoder.py
src/bcijelly/model/_lfads/lfads_torch/modules/initializers.py
src/bcijelly/model/_lfads/lfads_torch/modules/l2.py
src/bcijelly/model/_lfads/lfads_torch/modules/priors.py
src/bcijelly/model/_lfads/lfads_torch/modules/readin_readout.py
src/bcijelly/model/_lfads/lfads_torch/modules/recons.py
src/bcijelly/model/_lfads/lfads_torch/modules/recurrent.py
src/bcijelly/model/_lfads/lfads_torch/post_run/__init__.py
src/bcijelly/model/_lfads/lfads_torch/post_run/analysis.py
src/bcijelly/model/_lfads/lfads_torch/post_run/pbt.py
src/bcijelly/model/_lstm/__init__.py
src/bcijelly/model/_lstm/classifier.py
src/bcijelly/model/_lstm/plot_csv.py
src/bcijelly/model/_lstm/regressor.py
src/bcijelly/model/_lstm/shared_utils.py
src/bcijelly/model/_lstm/speech.py
src/bcijelly/model/_mfsnn/__init__.py
src/bcijelly/model/_mfsnn/classifier.py
src/bcijelly/model/_mfsnn/model.py
src/bcijelly/model/_mlp/__init__.py
src/bcijelly/model/_mlp/classifier.py
src/bcijelly/model/_mlp/regressor.py
src/bcijelly/model/_mscformer/__init__.py
src/bcijelly/model/_mscformer/classifier.py
src/bcijelly/model/_mscformer/model.py
src/bcijelly/model/_ndt1/__init__.py
src/bcijelly/model/_ndt1/classifier.py
src/bcijelly/model/_ndt1/data_process.py
src/bcijelly/model/_ndt1/plot_csv.py
src/bcijelly/model/_ndt1/regressor.py
src/bcijelly/model/_ndt1/shared_utils.py
src/bcijelly/model/_ndt1/models/__init__.py
src/bcijelly/model/_ndt1/models/config_utils.py
src/bcijelly/model/_ndt1/models/masker.py
src/bcijelly/model/_ndt1/models/model_output.py
src/bcijelly/model/_ndt1/models/ndt1.py
src/bcijelly/model/_ndt1/models/patcher.py
src/bcijelly/model/_ndt1/models/region_lookup.py
src/bcijelly/model/_ndt2/__init__.py
src/bcijelly/model/_ndt2/classifier.py
src/bcijelly/model/_ndt2/data_process.py
src/bcijelly/model/_ndt2/plot_csv.py
src/bcijelly/model/_ndt2/regressor.py
src/bcijelly/model/_ndt2/shared_utils.py
src/bcijelly/model/_ndt2/models/__init__.py
src/bcijelly/model/_ndt2/models/config_utils.py
src/bcijelly/model/_ndt2/models/masker.py
src/bcijelly/model/_ndt2/models/model_output.py
src/bcijelly/model/_ndt2/models/patcher.py
src/bcijelly/model/_ndt2/models/region_lookup.py
src/bcijelly/model/_ndt2/models/stpatch.py
src/bcijelly/model/_nomad/__init__.py
src/bcijelly/model/_nomad/classifier.py
src/bcijelly/model/_nomad/model.py
src/bcijelly/model/_nomad/regressor.py
src/bcijelly/model/_nomad/wrapper_utils.py
src/bcijelly/model/_nomad/lfads-torch/setup.py
src/bcijelly/model/_nomad/lfads-torch/lfads_torch/__init__.py
src/bcijelly/model/_nomad/lfads-torch/lfads_torch/callbacks.py
src/bcijelly/model/_nomad/lfads-torch/lfads_torch/datamodules.py
src/bcijelly/model/_nomad/lfads-torch/lfads_torch/metrics.py
src/bcijelly/model/_nomad/lfads-torch/lfads_torch/model.py
src/bcijelly/model/_nomad/lfads-torch/lfads_torch/run_model.py
src/bcijelly/model/_nomad/lfads-torch/lfads_torch/tuples.py
src/bcijelly/model/_nomad/lfads-torch/lfads_torch/utils.py
src/bcijelly/model/_nomad/lfads-torch/lfads_torch/extensions/__init__.py
src/bcijelly/model/_nomad/lfads-torch/lfads_torch/extensions/nlb.py
src/bcijelly/model/_nomad/lfads-torch/lfads_torch/extensions/tune.py
src/bcijelly/model/_nomad/lfads-torch/lfads_torch/modules/__init__.py
src/bcijelly/model/_nomad/lfads-torch/lfads_torch/modules/augmentations.py
src/bcijelly/model/_nomad/lfads-torch/lfads_torch/modules/decoder.py
src/bcijelly/model/_nomad/lfads-torch/lfads_torch/modules/encoder.py
src/bcijelly/model/_nomad/lfads-torch/lfads_torch/modules/initializers.py
src/bcijelly/model/_nomad/lfads-torch/lfads_torch/modules/l2.py
src/bcijelly/model/_nomad/lfads-torch/lfads_torch/modules/priors.py
src/bcijelly/model/_nomad/lfads-torch/lfads_torch/modules/readin_readout.py
src/bcijelly/model/_nomad/lfads-torch/lfads_torch/modules/recons.py
src/bcijelly/model/_nomad/lfads-torch/lfads_torch/modules/recurrent.py
src/bcijelly/model/_nomad/lfads-torch/lfads_torch/post_run/__init__.py
src/bcijelly/model/_nomad/lfads-torch/lfads_torch/post_run/analysis.py
src/bcijelly/model/_nomad/lfads-torch/lfads_torch/post_run/pbt.py
src/bcijelly/model/_poyo/__init__.py
src/bcijelly/model/_poyo/build_poyo_input.py
src/bcijelly/model/_poyo/classifier.py
src/bcijelly/model/_poyo/plot_csv.py
src/bcijelly/model/_poyo/regressor.py
src/bcijelly/model/_poyo/shared_utils.py
src/bcijelly/model/_seqvae/__init__.py
src/bcijelly/model/_seqvae/regressor.py
src/bcijelly/model/_stabilization/__init__.py
src/bcijelly/model/_stabilization/classifier.py
src/bcijelly/model/_stabilization/kalman_utils.py
src/bcijelly/model/_stabilization/regressor.py
src/bcijelly/model/_stabilization/shared_utils.py
src/bcijelly/model/_stabilization/stabilization_mlp.py
src/bcijelly/model/_stabilization/stabilization_mlp_regression.py
src/bcijelly/model/_transformer/__init__.py
src/bcijelly/model/_transformer/classifier.py
src/bcijelly/model/_transformer/plot_csv.py
src/bcijelly/model/_transformer/regressor.py
src/bcijelly/model/_transformer/shared_utils.py
src/bcijelly/model/_transformer/speech.py
src/bcijelly/model/_unibci/__init__.py
src/bcijelly/model/_unibci/classifier.py
src/bcijelly/model/_unibci/plot_csv.py
src/bcijelly/model/_unibci/regressor.py
src/bcijelly/model/_unibci/shared_utils.py
src/bcijelly/model/_unibci/tokenizer.py
src/bcijelly/model/_unibci/models/__init__.py
src/bcijelly/model/_unibci/models/model.py
src/bcijelly/model/_unibci/models/utils.py
src/bcijelly/utility/__init__.py
src/bcijelly/utility/summary.py
src/bcijelly/weights/__init__.py
src/bcijelly/weights/_types.py
tests/test_agent.py
tests/test_custom_data.py
tests/test_custom_params.py
tests/test_demo.py
tests/test_llm_search_agent.py
tests/test_load_data.py
tests/test_model.py