LICENSE
README.md
pyproject.toml
src/medlatents/__init__.py
src/medlatents/configs.py
src/medlatents/py.typed
src/medlatents.egg-info/PKG-INFO
src/medlatents.egg-info/SOURCES.txt
src/medlatents.egg-info/dependency_links.txt
src/medlatents.egg-info/requires.txt
src/medlatents.egg-info/top_level.txt
src/medlatents/autoregressive/__init__.py
src/medlatents/autoregressive/transformer.py
src/medlatents/autoregressive/speculative/__init__.py
src/medlatents/autoregressive/speculative/decoder.py
src/medlatents/autoregressive/speculative/eagle.py
src/medlatents/autoregressive/speculative/medusa.py
src/medlatents/bayesian_flow/__init__.py
src/medlatents/bayesian_flow/entropy.py
src/medlatents/bayesian_flow/guided_samplers.py
src/medlatents/bayesian_flow/loss.py
src/medlatents/bayesian_flow/schedules.py
src/medlatents/bayesian_flow/transformer.py
src/medlatents/bayesian_flow/solvers/__init__.py
src/medlatents/bayesian_flow/solvers/base.py
src/medlatents/bayesian_flow/solvers/dpm.py
src/medlatents/bayesian_flow/solvers/euler.py
src/medlatents/bayesian_flow/solvers/exponential.py
src/medlatents/common/__init__.py
src/medlatents/common/schedules.py
src/medlatents/common/time.py
src/medlatents/conditioning/__init__.py
src/medlatents/conditioning/bundle.py
src/medlatents/conditioning/collate.py
src/medlatents/conditioning/encoders.py
src/medlatents/config_models/__init__.py
src/medlatents/config_models/pydantic.py
src/medlatents/data/__init__.py
src/medlatents/data/tokenized_datasets.py
src/medlatents/data/tokenizers.py
src/medlatents/diffusion/__init__.py
src/medlatents/diffusion/continuous.py
src/medlatents/diffusion/d3pm.py
src/medlatents/diffusion/schedules.py
src/medlatents/diffusion/sedd.py
src/medlatents/evaluation/__init__.py
src/medlatents/evaluation/classifier_utility.py
src/medlatents/evaluation/factorial.py
src/medlatents/evaluation/features.py
src/medlatents/evaluation/fid.py
src/medlatents/evaluation/generative.py
src/medlatents/evaluation/memorization.py
src/medlatents/evaluation/reconstruction.py
src/medlatents/flow_matching/__init__.py
src/medlatents/flow_matching/continuous.py
src/medlatents/flow_matching/core.py
src/medlatents/flow_matching/discrete.py
src/medlatents/flow_matching/evaluate.py
src/medlatents/flow_matching/generate.py
src/medlatents/flow_matching/optimal_transport.py
src/medlatents/flow_matching/schedulers.py
src/medlatents/flow_matching/shortcut.py
src/medlatents/generation/__init__.py
src/medlatents/generation/generator.py
src/medlatents/inference/__init__.py
src/medlatents/inference/base.py
src/medlatents/inference/inpainting.py
src/medlatents/inference/kv_cache.py
src/medlatents/inference/quantization.py
src/medlatents/inference/super_resolution.py
src/medlatents/maskgit/__init__.py
src/medlatents/maskgit/transformer.py
src/medlatents/networks/__init__.py
src/medlatents/networks/diffusion_transformer.py
src/medlatents/networks/sprint.py
src/medlatents/networks/stability.py
src/medlatents/networks/transformer.py
src/medlatents/post_training/__init__.py
src/medlatents/post_training/configs.py
src/medlatents/post_training/distillation/__init__.py
src/medlatents/post_training/distillation/consistency.py
src/medlatents/post_training/distillation/reflow.py
src/medlatents/post_training/dpo/__init__.py
src/medlatents/post_training/dpo/base.py
src/medlatents/post_training/dpo/diffusion.py
src/medlatents/post_training/dpo/discrete.py
src/medlatents/post_training/dpo/flow.py
src/medlatents/post_training/dpo/spo.py
src/medlatents/post_training/preference/__init__.py
src/medlatents/post_training/preference/ai_feedback.py
src/medlatents/post_training/preference/data.py
src/medlatents/post_training/preference/reward_model.py
src/medlatents/post_training/rl/__init__.py
src/medlatents/post_training/rl/base.py
src/medlatents/post_training/rl/ddpo.py
src/medlatents/post_training/rl/gardo.py
src/medlatents/post_training/rl/grpo.py
src/medlatents/post_training/self_play/__init__.py
src/medlatents/post_training/self_play/iterative.py
src/medlatents/post_training/self_play/rejection.py
src/medlatents/rasterization/__init__.py
src/medlatents/rasterization/hilbert_curve.py
src/medlatents/rasterization/raster_scan.py
src/medlatents/rasterization/s_curve.py
src/medlatents/rasterization/zorder_curve.py
src/medlatents/sampling/__init__.py
src/medlatents/sampling/autoregressive.py
src/medlatents/sampling/bayesian_flow.py
src/medlatents/sampling/compile.py
src/medlatents/sampling/diffusion.py
src/medlatents/sampling/discretization.py
src/medlatents/sampling/flow_diffusion.py
src/medlatents/sampling/guidance.py
src/medlatents/sampling/schedules.py
src/medlatents/sampling/maskgit/__init__.py
src/medlatents/sampling/maskgit/compile.py
src/medlatents/sampling/maskgit/critic.py
src/medlatents/sampling/maskgit/klass.py
src/medlatents/sampling/maskgit/remasking.py
src/medlatents/sampling/maskgit/sampling.py
src/medlatents/sampling/maskgit/schedules.py
src/medlatents/training/__init__.py
src/medlatents/training/augmentation.py
src/medlatents/training/checkpointing.py
src/medlatents/training/conditional.py
src/medlatents/training/continuous.py
src/medlatents/training/curriculum.py
src/medlatents/training/dataloader.py
src/medlatents/training/discrete.py
src/medlatents/training/losses.py
src/medlatents/training/parallelism.py
src/medlatents/training/performance.py
src/medlatents/training/profiling.py
src/medlatents/training/repa.py
src/medlatents/training/repa_e.py
src/medlatents/training/schedulers.py
src/medlatents/utils/__init__.py
src/medlatents/utils/numerics.py
src/medlatents/utils/special_tokens.py
src/medlatents/utils/state_dict.py
tests/test_advanced_features.py
tests/test_autoregressive.py
tests/test_bayesian_flow.py
tests/test_common.py
tests/test_compile.py
tests/test_conditioning.py
tests/test_config_models.py
tests/test_continuous.py
tests/test_data.py
tests/test_diffusion.py
tests/test_diffusion_train_script.py
tests/test_dit.py
tests/test_evaluation.py
tests/test_flow_generation.py
tests/test_flow_matching.py
tests/test_flow_matching_extras.py
tests/test_generation.py
tests/test_hardware.py
tests/test_haste.py
tests/test_inference.py
tests/test_integration.py
tests/test_invariants.py
tests/test_maskgit.py
tests/test_numerics.py
tests/test_performance.py
tests/test_post_training_distillation.py
tests/test_post_training_dpo.py
tests/test_post_training_preference.py
tests/test_post_training_rl.py
tests/test_post_training_self_play.py
tests/test_properties_extended.py
tests/test_public_api.py
tests/test_rasterization.py
tests/test_repa.py
tests/test_sampling.py
tests/test_scale.py
tests/test_training.py
tests/test_vecor.py