LICENSE
README.md
THIRD_PARTY_NOTICES.txt
VERSION
pyproject.toml
nvsubquadratic/__init__.py
nvsubquadratic/lazy_config.py
nvsubquadratic.egg-info/PKG-INFO
nvsubquadratic.egg-info/SOURCES.txt
nvsubquadratic.egg-info/dependency_links.txt
nvsubquadratic.egg-info/requires.txt
nvsubquadratic.egg-info/top_level.txt
nvsubquadratic/modules/__init__.py
nvsubquadratic/modules/_channels_first_utils.py
nvsubquadratic/modules/attention.py
nvsubquadratic/modules/causal_conv1d.py
nvsubquadratic/modules/ckconv_nd.py
nvsubquadratic/modules/condition_mixer.py
nvsubquadratic/modules/distributed_depthwise_conv_nd.py
nvsubquadratic/modules/drop_path.py
nvsubquadratic/modules/film.py
nvsubquadratic/modules/grn.py
nvsubquadratic/modules/hyena_nd.py
nvsubquadratic/modules/kernels_nd.py
nvsubquadratic/modules/layer_scale.py
nvsubquadratic/modules/mamba_nd.py
nvsubquadratic/modules/masks_nd.py
nvsubquadratic/modules/mlp.py
nvsubquadratic/modules/patchify.py
nvsubquadratic/modules/position_encoding.py
nvsubquadratic/modules/residual_block.py
nvsubquadratic/modules/rms_norm.py
nvsubquadratic/modules/rms_norm_channel_first.py
nvsubquadratic/modules/schedulers.py
nvsubquadratic/modules/sequence_mixer.py
nvsubquadratic/modules/subq_ops_causal_conv1d.py
nvsubquadratic/modules/vit5_attention.py
nvsubquadratic/modules/vit5_hyena_adapter.py
nvsubquadratic/modules/vit5_residual_block.py
nvsubquadratic/networks/classification_resnet.py
nvsubquadratic/networks/general_purpose_resnet.py
nvsubquadratic/networks/vit5_classification.py
nvsubquadratic/networks/baselines/__init__.py
nvsubquadratic/networks/baselines/unet_convnext.py
nvsubquadratic/networks/baselines/unet_convnext_v2.py
nvsubquadratic/ops/__init__.py
nvsubquadratic/ops/causal_conv1d_custom.py
nvsubquadratic/ops/circular_fftconv.py
nvsubquadratic/ops/fftconv.py
nvsubquadratic/ops/fftconv_chunked.py
nvsubquadratic/ops/fftconv_custom.py
nvsubquadratic/ops/mixed_fftconv.py
nvsubquadratic/parallel/__init__.py
nvsubquadratic/parallel/a2a_comms.py
nvsubquadratic/parallel/utils.py
nvsubquadratic/testing/__init__.py
nvsubquadratic/testing/utils.py
nvsubquadratic/utils/init.py
nvsubquadratic/utils/qk_norm.py
nvsubquadratic/utils/quack_utils.py
nvsubquadratic/utils/rope.py
tests/test_autoregressive_wrapper.py
tests/test_basic.py
tests/test_checkpoint_resume.py
tests/test_classification_loss.py
tests/test_compile_lcache.py
tests/test_dali_epoch_transition.py
tests/test_dali_rand_augment.py
tests/test_h5_caching.py
tests/test_image_grid_callback.py
tests/test_iteration_speed_callback.py
tests/test_nightly_validation.py
tests/test_nightly_well_validation.py
tests/test_param_groups.py
tests/test_pixel_scaling.py
tests/test_safe_eval_resolver.py
tests/test_well_dataloader_parity.py
tests/test_well_staging_ddp.py