MANIFEST.in
pyproject.toml
setup.py
areno/__init__.py
areno.egg-info/PKG-INFO
areno.egg-info/SOURCES.txt
areno.egg-info/dependency_links.txt
areno.egg-info/entry_points.txt
areno.egg-info/requires.txt
areno.egg-info/top_level.txt
areno/accel/__init__.py
areno/accel/_extension.py
areno/accel/activations.py
areno/accel/conv.py
areno/accel/embedding.py
areno/accel/linear.py
areno/accel/moe.py
areno/accel/normalization.py
areno/accel/ops.py
areno/accel/router.py
areno/accel/routing.py
areno/accel/topk.py
areno/accel/csrc/activation.cu
areno/accel/csrc/atomic_utils.cuh
areno/accel/csrc/conv.cu
areno/accel/csrc/embedding.cu
areno/accel/csrc/extension.cpp
areno/accel/csrc/linear.cu
areno/accel/csrc/moe_align_kernel.cu
areno/accel/csrc/moe_permute.cu
areno/accel/csrc/normalization.cu
areno/accel/csrc/router.cu
areno/accel/csrc/topk.cu
areno/accel/kernels/__init__.py
areno/accel/kernels/fused_moe.py
areno/accel/kernels/group_rmsnorm.py
areno/accel/kernels/seg_la.py
areno/api/__init__.py
areno/api/advantages.py
areno/api/algorithms.py
areno/api/config.py
areno/api/context.py
areno/api/data.py
areno/api/data_utils.py
areno/api/defaults.py
areno/api/metrics.py
areno/api/models.py
areno/api/rewards.py
areno/api/roles.py
areno/api/tokenizer.py
areno/api/trainer.py
areno/api/trainer_config.py
areno/api/trainer_factory.py
areno/api/backend/__init__.py
areno/api/backend/base.py
areno/api/backend/areno/__init__.py
areno/api/backend/areno/backend.py
areno/api/loss_fns/__init__.py
areno/api/loss_fns/dpo.py
areno/api/loss_fns/grpo.py
areno/api/loss_fns/gspo.py
areno/api/loss_fns/layout.py
areno/api/loss_fns/ppo.py
areno/api/loss_fns/sft.py
areno/api/trainers/__init__.py
areno/api/trainers/dpo.py
areno/api/trainers/policy_only.py
areno/api/trainers/ppo.py
areno/api/trainers/sft.py
areno/cli/__init__.py
areno/cli/main.py
areno/cli/model_refs.py
areno/cli/serve.py
areno/cli/train.py
areno/engine/__init__.py
areno/engine/api.py
areno/engine/config.py
areno/engine/inference.py
areno/engine/log.py
areno/engine/modeling.py
areno/engine/protocol.py
areno/engine/roles.py
areno/engine/training.py
areno/engine/worker.py
areno/engine/checkpoints/__init__.py
areno/engine/checkpoints/common.py
areno/engine/checkpoints/io.py
areno/engine/data/__init__.py
areno/engine/data/batch.py
areno/engine/data/rollout_state.py
areno/engine/data/sampling.py
areno/engine/data/tokenizer.py
areno/engine/layers/__init__.py
areno/engine/layers/attention.py
areno/engine/layers/linear.py
areno/engine/layers/mlp.py
areno/engine/layers/norm.py
areno/engine/layers/rotary.py
areno/engine/layers/vocab.py
areno/engine/layers/attention_backend/__init__.py
areno/engine/layers/attention_backend/common.py
areno/engine/layers/attention_backend/infer.py
areno/engine/layers/attention_backend/train.py
areno/engine/optim/__init__.py
areno/engine/optim/adamw_8bit.py
areno/engine/optim/adamw_fp32_master.py
areno/engine/parallel/__init__.py
areno/engine/parallel/collectives.py
areno/engine/parallel/context.py
areno/engine/runtime/__init__.py
areno/engine/runtime/common.py
areno/engine/runtime/decode_graph.py
areno/engine/runtime/logprobs.py
areno/engine/runtime/metadata.py
areno/engine/runtime/recompute.py
areno/engine/runtime/rollout.py
areno/engine/runtime/train_step.py
areno/experimental/__init__.py
areno/models/__init__.py
areno/models/base.py
areno/models/registry.py
areno/models/_shared/__init__.py
areno/models/_shared/dynamo_wrappers.py
areno/models/bailing/__init__.py
areno/models/bailing/checkpoint.py
areno/models/bailing/model.py
areno/models/gemma4/__init__.py
areno/models/gemma4/checkpoint.py
areno/models/gemma4/model.py
areno/models/llama/__init__.py
areno/models/llama/checkpoint.py
areno/models/llama/model.py
areno/models/minicpmv46/__init__.py
areno/models/minicpmv46/checkpoint.py
areno/models/minicpmv46/model.py
areno/models/qwen3/__init__.py
areno/models/qwen3/checkpoint.py
areno/models/qwen3/model.py
areno/models/qwen3_5/__init__.py
areno/models/qwen3_5/checkpoint.py
areno/models/qwen3_5/model.py
tests/test_algorithms_cpu.py
tests/test_cli_model_refs_cpu.py
tests/test_config_data_cpu.py
tests/test_logprobs_cpu.py
tests/test_losses_rewards_cpu.py
tests/test_metrics_cpu.py
tests/test_more_losses_cpu.py
tests/test_protocol_cpu.py
tests/test_recompute_cpu.py
tests/test_registry_cpu.py
tests/test_runtime_utils_cpu.py
tests/test_sampling_cpu.py
tests/test_tokenizer_api_cpu.py
tests/test_trainer_api_cpu.py
tests/test_trainer_dataset_utils_cpu.py