LICENSE
README.md
pyproject.toml
setup.py
unirl/__init__.py
unirl/train_ar.py
unirl/train_diffusion.py
unirl/train_pe.py
unirl/train_unified_model.py
unirl.egg-info/PKG-INFO
unirl.egg-info/SOURCES.txt
unirl.egg-info/dependency_links.txt
unirl.egg-info/requires.txt
unirl.egg-info/top_level.txt
unirl/algorithms/__init__.py
unirl/algorithms/base.py
unirl/algorithms/diffusionnft.py
unirl/algorithms/drpo.py
unirl/algorithms/flowdppo.py
unirl/algorithms/flowgrpo.py
unirl/algorithms/grpo.py
unirl/algorithms/normalizers.py
unirl/config/__init__.py
unirl/config/require.py
unirl/config/validation.py
unirl/data/__init__.py
unirl/data/data_source.py
unirl/data/datasets.py
unirl/distributed/__init__.py
unirl/distributed/utils.py
unirl/distributed/group/device_pool.py
unirl/distributed/group/dispatch.py
unirl/distributed/group/handle.py
unirl/distributed/group/placement.py
unirl/distributed/group/remote.py
unirl/distributed/group/worker.py
unirl/distributed/tensor/__init__.py
unirl/distributed/tensor/batch.py
unirl/distributed/tensor/factory.py
unirl/distributed/tensor/grad_context.py
unirl/distributed/tensor/pytree.py
unirl/distributed/tensor/transport.py
unirl/distributed/tensor/backend/__init__.py
unirl/distributed/tensor/backend/colocate_store/__init__.py
unirl/distributed/tensor/backend/colocate_store/handle.py
unirl/distributed/tensor/backend/colocate_store/store.py
unirl/distributed/tensor/backend/colocate_store/transport.py
unirl/distributed/tensor/backend/gpu_store/__init__.py
unirl/distributed/tensor/backend/gpu_store/handle.py
unirl/distributed/tensor/backend/gpu_store/transport.py
unirl/distributed/tensor/backend/gpu_store/worker.py
unirl/distributed/tensor/backend/transfer_queue/__init__.py
unirl/distributed/tensor/backend/transfer_queue/base.py
unirl/distributed/tensor/backend/transfer_queue/mooncake.py
unirl/distributed/tensor/backend/transfer_queue/runtime.py
unirl/distributed/tensor/backend/transfer_queue/simple.py
unirl/distributed/tensor/backend/transfer_queue/topology.py
unirl/distributed/tensor/backend/transfer_queue/transport.py
unirl/distributed/weight_sync/__init__.py
unirl/distributed/weight_sync/payload.py
unirl/distributed/weight_sync/full/__init__.py
unirl/distributed/weight_sync/full/base.py
unirl/distributed/weight_sync/full/ipc.py
unirl/distributed/weight_sync/full/nccl.py
unirl/distributed/weight_sync/full/tensor.py
unirl/distributed/weight_sync/lora/__init__.py
unirl/distributed/weight_sync/lora/base.py
unirl/distributed/weight_sync/lora/local.py
unirl/distributed/weight_sync/lora/remote.py
unirl/models/__init__.py
unirl/models/bagel/__init__.py
unirl/models/bagel/bundle.py
unirl/models/bagel/conditions.py
unirl/models/bagel/config.py
unirl/models/bagel/diffusion.py
unirl/models/bagel/pipeline.py
unirl/models/bagel/rl_ops.py
unirl/models/bagel/vae.py
unirl/models/bagel/vendor/__init__.py
unirl/models/bagel/vendor/inferencer.py
unirl/models/bagel/vendor/data/__init__.py
unirl/models/bagel/vendor/data/data_utils.py
unirl/models/bagel/vendor/data/transforms.py
unirl/models/bagel/vendor/modeling/__init__.py
unirl/models/bagel/vendor/modeling/autoencoder.py
unirl/models/bagel/vendor/modeling/bagel/__init__.py
unirl/models/bagel/vendor/modeling/bagel/bagel.py
unirl/models/bagel/vendor/modeling/bagel/modeling_utils.py
unirl/models/bagel/vendor/modeling/bagel/qwen2_navit.py
unirl/models/bagel/vendor/modeling/bagel/siglip_navit.py
unirl/models/bagel/vendor/modeling/cache_utils/__init__.py
unirl/models/bagel/vendor/modeling/cache_utils/taylorseer.py
unirl/models/bagel/vendor/modeling/qwen2/__init__.py
unirl/models/bagel/vendor/modeling/qwen2/configuration_qwen2.py
unirl/models/bagel/vendor/modeling/qwen2/modeling_qwen2.py
unirl/models/bagel/vendor/modeling/qwen2/tokenization_qwen2.py
unirl/models/bagel/vendor/modeling/qwen2/tokenization_qwen2_fast.py
unirl/models/bagel/vendor/modeling/siglip/__init__.py
unirl/models/bagel/vendor/modeling/siglip/configuration_siglip.py
unirl/models/bagel/vendor/modeling/siglip/convert_siglip_to_hf.py
unirl/models/bagel/vendor/modeling/siglip/image_processing_siglip.py
unirl/models/bagel/vendor/modeling/siglip/modeling_siglip.py
unirl/models/bagel/vendor/modeling/siglip/processing_siglip.py
unirl/models/bagel/vendor/modeling/siglip/tokenization_siglip.py
unirl/models/flux2_klein/__init__.py
unirl/models/flux2_klein/bundle.py
unirl/models/flux2_klein/conditions.py
unirl/models/flux2_klein/config.py
unirl/models/flux2_klein/diffusion.py
unirl/models/flux2_klein/flux2_klein_utils.py
unirl/models/flux2_klein/pipeline.py
unirl/models/flux2_klein/schedule.py
unirl/models/flux2_klein/text_embed.py
unirl/models/flux2_klein/vae.py
unirl/models/hunyuan_image3/__init__.py
unirl/models/hunyuan_image3/ar.py
unirl/models/hunyuan_image3/bundle.py
unirl/models/hunyuan_image3/conditions.py
unirl/models/hunyuan_image3/config.py
unirl/models/hunyuan_image3/diffusion.py
unirl/models/hunyuan_image3/diffusion_state.py
unirl/models/hunyuan_image3/pipeline.py
unirl/models/hunyuan_image3/text_embed.py
unirl/models/hunyuan_image3/vae.py
unirl/models/hunyuan_image3/vit_encode.py
unirl/models/hunyuan_image3/modes/__init__.py
unirl/models/hunyuan_image3/modes/i2t.py
unirl/models/hunyuan_image3/modes/it2i.py
unirl/models/hunyuan_image3/modes/t2i.py
unirl/models/hunyuan_image3/modes/t2t.py
unirl/models/hunyuan_video/__init__.py
unirl/models/hunyuan_video/bundle.py
unirl/models/hunyuan_video/conditions.py
unirl/models/hunyuan_video/config.py
unirl/models/hunyuan_video/diffusion.py
unirl/models/hunyuan_video/pipeline.py
unirl/models/hunyuan_video/text_embed.py
unirl/models/hunyuan_video/vae.py
unirl/models/hunyuan_video15/__init__.py
unirl/models/hunyuan_video15/bundle.py
unirl/models/hunyuan_video15/conditions.py
unirl/models/hunyuan_video15/config.py
unirl/models/hunyuan_video15/diffusion.py
unirl/models/hunyuan_video15/pipeline.py
unirl/models/hunyuan_video15/text_embed.py
unirl/models/hunyuan_video15/vae.py
unirl/models/pe/__init__.py
unirl/models/pe/bundle.py
unirl/models/pe/pipeline.py
unirl/models/qwen3/__init__.py
unirl/models/qwen3/ar.py
unirl/models/qwen3/bundle.py
unirl/models/qwen3/chat_template.py
unirl/models/qwen3/conditions.py
unirl/models/qwen3/config.py
unirl/models/qwen3/pipeline.py
unirl/models/qwen_image/__init__.py
unirl/models/qwen_image/bundle.py
unirl/models/qwen_image/conditions.py
unirl/models/qwen_image/config.py
unirl/models/qwen_image/diffusion.py
unirl/models/qwen_image/pipeline.py
unirl/models/qwen_image/text_embed.py
unirl/models/qwen_image/vae.py
unirl/models/qwen_vl/__init__.py
unirl/models/qwen_vl/ar.py
unirl/models/qwen_vl/bundle.py
unirl/models/qwen_vl/chat_template.py
unirl/models/qwen_vl/conditions.py
unirl/models/qwen_vl/config.py
unirl/models/qwen_vl/pipeline.py
unirl/models/sd3/__init__.py
unirl/models/sd3/bundle.py
unirl/models/sd3/conditions.py
unirl/models/sd3/config.py
unirl/models/sd3/diffusion.py
unirl/models/sd3/pipeline.py
unirl/models/sd3/text_embed.py
unirl/models/sd3/vae.py
unirl/models/types/__init__.py
unirl/models/types/ar.py
unirl/models/types/bundle.py
unirl/models/types/codec.py
unirl/models/types/diffusion.py
unirl/models/types/embedding.py
unirl/models/types/pipeline.py
unirl/models/types/replay_result.py
unirl/models/wan21/__init__.py
unirl/models/wan21/bundle.py
unirl/models/wan21/clip_vision_encode.py
unirl/models/wan21/conditions.py
unirl/models/wan21/config.py
unirl/models/wan21/diffusion.py
unirl/models/wan21/image_encode.py
unirl/models/wan21/pipeline.py
unirl/models/wan21/text_embed.py
unirl/models/wan21/vae.py
unirl/models/wan22/__init__.py
unirl/models/wan22/bundle.py
unirl/models/wan22/config.py
unirl/models/wan22/diffusion.py
unirl/models/wan22/pipeline.py
unirl/reward/__init__.py
unirl/reward/base.py
unirl/reward/remote.py
unirl/reward/service.py
unirl/reward/local/__init__.py
unirl/reward/local/aesthetic.py
unirl/reward/local/base.py
unirl/reward/local/clip.py
unirl/reward/local/device.py
unirl/reward/local/geneval2.py
unirl/reward/local/gsm8k_exact_match.py
unirl/reward/local/hpsv2.py
unirl/reward/local/hpsv3.py
unirl/reward/local/image_reward.py
unirl/reward/local/math_boxed.py
unirl/reward/local/mathverify.py
unirl/reward/local/mc_exact_match.py
unirl/reward/local/ocr.py
unirl/reward/local/pickscore.py
unirl/reward/local/registry.py
unirl/reward/local/video.py
unirl/reward/local/video_pickscore.py
unirl/rollout/__init__.py
unirl/rollout/engine/__init__.py
unirl/rollout/engine/base.py
unirl/rollout/engine/sigma_verify.py
unirl/rollout/engine/composed/__init__.py
unirl/rollout/engine/composed/config.py
unirl/rollout/engine/composed/engine.py
unirl/rollout/engine/sglang/__init__.py
unirl/rollout/engine/sglang/_sample_decode.py
unirl/rollout/engine/sglang/_text_fusion.py
unirl/rollout/engine/sglang/config.py
unirl/rollout/engine/sglang/engine.py
unirl/rollout/engine/sglang/request.py
unirl/rollout/engine/sglang/response.py
unirl/rollout/engine/sglang/_patches/__init__.py
unirl/rollout/engine/sglang/_patches/hijack.py
unirl/rollout/engine/sglang/_patches/io_struct.py
unirl/rollout/engine/sglang/_patches/lora_req.py
unirl/rollout/engine/sglang/_patches/memory_saver.py
unirl/rollout/engine/sglang/_patches/patch_conditions.py
unirl/rollout/engine/sglang/_patches/patch_dance.py
unirl/rollout/engine/sglang/_patches/patch_denoising.py
unirl/rollout/engine/sglang/_patches/patch_gpu_worker.py
unirl/rollout/engine/sglang/_patches/patch_grouped_dispatch.py
unirl/rollout/engine/sglang/_patches/patch_latent_prep.py
unirl/rollout/engine/sglang/_patches/patch_lora_slice_2d.py
unirl/rollout/engine/sglang/_patches/patch_lora_tensors.py
unirl/rollout/engine/sglang/_patches/patch_pipeline.py
unirl/rollout/engine/sglang/_patches/patch_platform_device.py
unirl/rollout/engine/sglang/_patches/patch_rollout_trajectory.py
unirl/rollout/engine/sglang/_patches/patch_sampling_io.py
unirl/rollout/engine/sglang/_patches/patch_scheduler.py
unirl/rollout/engine/sglang/_patches/patch_sd3_lora_pipeline.py
unirl/rollout/engine/sglang/_patches/patch_set_timesteps.py
unirl/rollout/engine/sglang/_patches/patch_srt.py
unirl/rollout/engine/sglang/_patches/patch_vae_decode_safe.py
unirl/rollout/engine/sglang/_patches/patch_weights_updater.py
unirl/rollout/engine/sglang_llm/__init__.py
unirl/rollout/engine/sglang_llm/_server.py
unirl/rollout/engine/sglang_llm/config.py
unirl/rollout/engine/sglang_llm/engine.py
unirl/rollout/engine/trainside/__init__.py
unirl/rollout/engine/trainside/config.py
unirl/rollout/engine/trainside/engine.py
unirl/rollout/engine/vllm_omni/__init__.py
unirl/rollout/engine/vllm_omni/config.py
unirl/rollout/engine/vllm_omni/engine.py
unirl/rollout/engine/vllm_omni/request.py
unirl/rollout/engine/vllm_omni/response.py
unirl/rollout/engine/vllm_omni/vllm_patches.py
unirl/rollout/engine/vllm_omni/_shared/__init__.py
unirl/rollout/engine/vllm_omni/_shared/flow_match_sde_scheduler.py
unirl/rollout/engine/vllm_omni/compat/__init__.py
unirl/rollout/engine/vllm_omni/compat/hi3_lora.py
unirl/rollout/engine/vllm_omni/compat/tokenizer.py
unirl/rollout/engine/vllm_omni/hi3/__init__.py
unirl/rollout/engine/vllm_omni/hi3/ar_capture.py
unirl/rollout/engine/vllm_omni/hi3/pipeline.py
unirl/rollout/engine/vllm_omni/hi3/sde_scheduler.py
unirl/rollout/engine/vllm_omni/hv15/__init__.py
unirl/rollout/engine/vllm_omni/hv15/pipeline.py
unirl/rollout/engine/vllm_omni/sd3/__init__.py
unirl/rollout/engine/vllm_omni/sd3/pipeline.py
unirl/rollout/engine/vllm_omni/weight_sync/__init__.py
unirl/rollout/engine/vllm_omni/weight_sync/ar_extension.py
unirl/rollout/engine/vllm_omni/weight_sync/bucketed_transfer.py
unirl/rollout/engine/vllm_omni/weight_sync/checksum.py
unirl/rollout/engine/vllm_omni/weight_sync/dit_extension.py
unirl/rollout/engine/vllm_omni/weight_sync/ipc_dispatch.py
unirl/rollout/engine/vllm_omni/weight_sync/ipc_receive_mixin.py
unirl/rollout/engine/vllm_omni/weight_sync/nccl_receive_mixin.py
unirl/rollout/engine/vllm_omni/weight_sync/sgl_compat.py
unirl/sde/__init__.py
unirl/sde/kernels.py
unirl/sde/noise.py
unirl/sde/rules.py
unirl/sde/runtime.py
unirl/train/__init__.py
unirl/train/configs.py
unirl/train/ema.py
unirl/train/factories.py
unirl/train/fsdp_utils.py
unirl/train/inject.py
unirl/train/shadow.py
unirl/train/stack.py
unirl/train/unified_model_stack.py
unirl/train/backend/__init__.py
unirl/train/backend/base.py
unirl/train/backend/fsdp.py
unirl/train/backend/protocols.py
unirl/trainer/ar.py
unirl/trainer/base.py
unirl/trainer/diffusion.py
unirl/trainer/pe.py
unirl/trainer/unified_model.py
unirl/types/__init__.py
unirl/types/batch_ops.py
unirl/types/media.py
unirl/types/media_preview.py
unirl/types/noise_recipe.py
unirl/types/primitives.py
unirl/types/prompts.py
unirl/types/reward.py
unirl/types/rollout_req.py
unirl/types/rollout_resp.py
unirl/types/sampling.py
unirl/types/trajectory_store.py
unirl/types/conditions/__init__.py
unirl/types/conditions/base.py
unirl/types/conditions/fused_multimodal.py
unirl/types/conditions/image.py
unirl/types/conditions/text.py
unirl/types/segments/__init__.py
unirl/types/segments/base.py
unirl/types/segments/latent.py
unirl/types/segments/text.py
unirl/utils/__init__.py
unirl/utils/adapter_utils.py
unirl/utils/distributed_utils.py
unirl/utils/dtypes.py
unirl/utils/hydra.py
unirl/utils/media.py
unirl/utils/misc.py
unirl/utils/peft_merge.py
unirl/utils/prepare_dapo_math.py
unirl/utils/prepare_geo3k_mc.py
unirl/utils/scheduler_utils.py
unirl/utils/sglang_endpoint.py
unirl/utils/timing.py
unirl/utils/wandb_logger.py
unirl/utils/wandb_metrics.py