LICENSE
README.md
pyproject.toml
src/utils.py
src/omar_rq/__init__.py
src/omar_rq/callbacks.py
src/omar_rq/compute_input_stats.py
src/omar_rq/cosineannealingscheduler.py
src/omar_rq/downstream.py
src/omar_rq/get_model.py
src/omar_rq/predict.py
src/omar_rq/train.py
src/omar_rq/utils.py
src/omar_rq.egg-info/PKG-INFO
src/omar_rq.egg-info/SOURCES.txt
src/omar_rq.egg-info/dependency_links.txt
src/omar_rq.egg-info/requires.txt
src/omar_rq.egg-info/top_level.txt
src/omar_rq/data/__init__.py
src/omar_rq/data/data_utils.py
src/omar_rq/data/discotube.py
src/omar_rq/data/discotube_text_audio.py
src/omar_rq/modules/__init__.py
src/omar_rq/modules/classifier.py
src/omar_rq/modules/codebooks.py
src/omar_rq/modules/finite_scalar_quantizer.py
src/omar_rq/modules/maskingmodel.py
src/omar_rq/modules/simclr.py
src/omar_rq/nets/__init__.py
src/omar_rq/nets/common_former.py
src/omar_rq/nets/conformer.py
src/omar_rq/nets/cqt.py
src/omar_rq/nets/encodec.py
src/omar_rq/nets/melspectrogram.py
src/omar_rq/nets/mlp.py
src/omar_rq/nets/net.py
src/omar_rq/nets/rope.py
src/omar_rq/nets/transformer.py
src/omar_rq/nets/waveform.py
src/omar_rq/prediction/callbacks.py
src/omar_rq/prediction/dataset.py
src/omar_rq/probe/data/__init__.py
src/omar_rq/probe/data/magnatagatune.py
src/omar_rq/probe/data/nsynth_pitch.py
src/omar_rq/probe/modules/__init__.py
src/omar_rq/probe/modules/sequence_classifiers.py
tests/test_omar_rq.py