MANIFEST.in
Readme.md
pyproject.toml
vitax_rag/__init__.py
vitax_rag/cli.py
vitax_rag/tree.py
vitax_rag/utils.py
vitax_rag.egg-info/PKG-INFO
vitax_rag.egg-info/SOURCES.txt
vitax_rag.egg-info/dependency_links.txt
vitax_rag.egg-info/entry_points.txt
vitax_rag.egg-info/requires.txt
vitax_rag.egg-info/top_level.txt
vitax_rag/lca/tree.py
vitax_rag/model/model_hyena.py
vitax_rag/model/model_vitax.py
vitax_rag/model/tokenizer_hyena.py
vitax_rag/model/hyena/huggingface.py
vitax_rag/model/hyena/standalone_hyenadna.py
vitax_rag/model/hyena/train.py
vitax_rag/model/hyena/csrc/fftconv/launch_fftconv.py
vitax_rag/model/hyena/csrc/fftconv/lut_code_gen.py
vitax_rag/model/hyena/csrc/fftconv/setup.py
vitax_rag/model/hyena/evals/hg38_inference.py
vitax_rag/model/hyena/evals/hg38_inference_decoder.py
vitax_rag/model/hyena/evals/instruction_tuned_genomics.py
vitax_rag/model/hyena/evals/soft_prompting_genomics.py
vitax_rag/model/hyena/src/callbacks/gpu_affinity.py
vitax_rag/model/hyena/src/callbacks/norms.py
vitax_rag/model/hyena/src/callbacks/params.py
vitax_rag/model/hyena/src/callbacks/progressive_resizing.py
vitax_rag/model/hyena/src/callbacks/seqlen_warmup_reload.py
vitax_rag/model/hyena/src/callbacks/timer.py
vitax_rag/model/hyena/src/callbacks/wandb.py
vitax_rag/model/hyena/src/dataloaders/__init__.py
vitax_rag/model/hyena/src/dataloaders/base.py
vitax_rag/model/hyena/src/dataloaders/et.py
vitax_rag/model/hyena/src/dataloaders/fault_tolerant_sampler.py
vitax_rag/model/hyena/src/dataloaders/genomics.py
vitax_rag/model/hyena/src/dataloaders/datasets/chromatin_profile_dataset.py
vitax_rag/model/hyena/src/dataloaders/datasets/genomic_bench_dataset.py
vitax_rag/model/hyena/src/dataloaders/datasets/hg38_char_tokenizer.py
vitax_rag/model/hyena/src/dataloaders/datasets/hg38_dataset.py
vitax_rag/model/hyena/src/dataloaders/datasets/hg38_fixed_dataset.py
vitax_rag/model/hyena/src/dataloaders/datasets/hg38_icl_dataset.py
vitax_rag/model/hyena/src/dataloaders/datasets/icl_genomics_dataset.py
vitax_rag/model/hyena/src/dataloaders/datasets/lm_dataset.py
vitax_rag/model/hyena/src/dataloaders/datasets/nucleotide_transformer_dataset.py
vitax_rag/model/hyena/src/dataloaders/datasets/species_dataset.py
vitax_rag/model/hyena/src/dataloaders/utils/vocabulary.py
vitax_rag/model/hyena/src/models/nn/__init__.py
vitax_rag/model/hyena/src/models/nn/adaptive_softmax.py
vitax_rag/model/hyena/src/models/nn/components.py
vitax_rag/model/hyena/src/models/nn/dxt.py
vitax_rag/model/hyena/src/models/nn/gate.py
vitax_rag/model/hyena/src/models/nn/residual.py
vitax_rag/model/hyena/src/models/nn/utils.py
vitax_rag/model/hyena/src/models/sequence/__init__.py
vitax_rag/model/hyena/src/models/sequence/base.py
vitax_rag/model/hyena/src/models/sequence/block.py
vitax_rag/model/hyena/src/models/sequence/block_fft.py
vitax_rag/model/hyena/src/models/sequence/dna_embedding.py
vitax_rag/model/hyena/src/models/sequence/ff.py
vitax_rag/model/hyena/src/models/sequence/hyena.py
vitax_rag/model/hyena/src/models/sequence/long_conv.py
vitax_rag/model/hyena/src/models/sequence/long_conv_kernel.py
vitax_rag/model/hyena/src/models/sequence/long_conv_lm.py
vitax_rag/model/hyena/src/models/sequence/mha.py
vitax_rag/model/hyena/src/models/sequence/model.py
vitax_rag/model/hyena/src/models/sequence/pool.py
vitax_rag/model/hyena/src/models/sequence/simple_lm.py
vitax_rag/model/hyena/src/ops/fftconv.py
vitax_rag/model/hyena/src/ops/krylov.py
vitax_rag/model/hyena/src/ops/toeplitz.py
vitax_rag/model/hyena/src/ops/unroll.py
vitax_rag/model/hyena/src/ops/vandermonde.py
vitax_rag/model/hyena/src/tasks/decoders.py
vitax_rag/model/hyena/src/tasks/encoders.py
vitax_rag/model/hyena/src/tasks/metrics.py
vitax_rag/model/hyena/src/tasks/tasks.py
vitax_rag/model/hyena/src/tasks/torchmetrics.py
vitax_rag/model/hyena/src/utils/__init__.py
vitax_rag/model/hyena/src/utils/config.py
vitax_rag/model/hyena/src/utils/distributed.py
vitax_rag/model/hyena/src/utils/optim_groups.py
vitax_rag/model/hyena/src/utils/permutations.py
vitax_rag/model/hyena/src/utils/profiling.py
vitax_rag/model/hyena/src/utils/registry.py
vitax_rag/model/hyena/src/utils/train.py
vitax_rag/model/hyena/src/utils/optim/lamb.py
vitax_rag/model/hyena/src/utils/optim/schedulers.py