├── .gitignore ├── .gitmodules ├── .pre-commit-config.yaml ├── LICENSE ├── LICENSE_NOTE.md ├── README.md ├── WinningModelDocumentationGuidelines.pdf ├── cfgs ├── dpsk-qwen-14b-awq.yaml ├── dpsk-qwen-14b-finetune-v1-epoch2-awq.yaml ├── dpsk-qwen-14b-finetune-v1-epoch2.yaml ├── dpsk-qwen-14b-finetune-v1-epoch4-awq-32codeprompt.yaml ├── dpsk-qwen-14b-finetune-v1-epoch4-awq-32codepromptv2.yaml ├── dpsk-qwen-14b-finetune-v1-epoch4-awq-diversesysprompt.yaml ├── dpsk-qwen-14b-finetune-v1-epoch4-awq.yaml ├── dpsk-qwen-14b-finetune-v1-epoch4-repkv-awq-kv4.yaml ├── dpsk-qwen-14b-finetune-v1-epoch4-repkv-awq.yaml ├── dpsk-qwen-14b-finetune-v1-epoch4.yaml └── dpsk-qwen-14b.yaml ├── data ├── aime_2025_30.csv ├── reference.csv └── stub.csv ├── directory_structure.txt ├── entry_points.md ├── figs ├── dpo_result.png ├── early_stop_analysis.jpg ├── inference_workflow.png ├── initialization_com.png ├── output_speed.png ├── sft_accuracy.png ├── sft_output_len.png ├── throughput_com.png └── token_length.png ├── imagination_aimo2 ├── __init__.py ├── aggregators.py ├── backup_submission.ipynb ├── local_eval.py └── local_eval_kaggle.py ├── kaggle_evaluation ├── __init__.py ├── aimo_2_gateway.py ├── aimo_2_inference_server.py └── core │ ├── __init__.py │ ├── base_gateway.py │ ├── generated │ ├── __init__.py │ ├── kaggle_evaluation_pb2.py │ └── kaggle_evaluation_pb2_grpc.py │ ├── kaggle_evaluation.proto │ ├── relay.py │ └── templates.py ├── kaggle_forum.md ├── pyproject.toml ├── requirements.txt ├── scripts ├── analyze_early_stop.py ├── analyze_llm_vote.py ├── deduplicate_sft_data.py ├── filter_dpo_data.py ├── plot_table_local.py ├── quant_awq.py ├── quant_rep_kv.py ├── run_cfg.sh ├── run_convert.sh ├── run_deduplicate.sh ├── run_dpo.sh ├── run_filter.sh ├── run_quant_awq.sh ├── run_quant_rep_kv.sh └── run_sft.sh └── settings.json /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/imagination-research/aimo2/HEAD/.gitignore -------------------------------------------------------------------------------- /.gitmodules: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/imagination-research/aimo2/HEAD/.gitmodules -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/imagination-research/aimo2/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/imagination-research/aimo2/HEAD/LICENSE -------------------------------------------------------------------------------- /LICENSE_NOTE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/imagination-research/aimo2/HEAD/LICENSE_NOTE.md -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/imagination-research/aimo2/HEAD/README.md -------------------------------------------------------------------------------- /WinningModelDocumentationGuidelines.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/imagination-research/aimo2/HEAD/WinningModelDocumentationGuidelines.pdf -------------------------------------------------------------------------------- /cfgs/dpsk-qwen-14b-awq.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/imagination-research/aimo2/HEAD/cfgs/dpsk-qwen-14b-awq.yaml -------------------------------------------------------------------------------- /cfgs/dpsk-qwen-14b-finetune-v1-epoch2-awq.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/imagination-research/aimo2/HEAD/cfgs/dpsk-qwen-14b-finetune-v1-epoch2-awq.yaml -------------------------------------------------------------------------------- /cfgs/dpsk-qwen-14b-finetune-v1-epoch2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/imagination-research/aimo2/HEAD/cfgs/dpsk-qwen-14b-finetune-v1-epoch2.yaml -------------------------------------------------------------------------------- /cfgs/dpsk-qwen-14b-finetune-v1-epoch4-awq-32codeprompt.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/imagination-research/aimo2/HEAD/cfgs/dpsk-qwen-14b-finetune-v1-epoch4-awq-32codeprompt.yaml -------------------------------------------------------------------------------- /cfgs/dpsk-qwen-14b-finetune-v1-epoch4-awq-32codepromptv2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/imagination-research/aimo2/HEAD/cfgs/dpsk-qwen-14b-finetune-v1-epoch4-awq-32codepromptv2.yaml -------------------------------------------------------------------------------- /cfgs/dpsk-qwen-14b-finetune-v1-epoch4-awq-diversesysprompt.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/imagination-research/aimo2/HEAD/cfgs/dpsk-qwen-14b-finetune-v1-epoch4-awq-diversesysprompt.yaml -------------------------------------------------------------------------------- /cfgs/dpsk-qwen-14b-finetune-v1-epoch4-awq.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/imagination-research/aimo2/HEAD/cfgs/dpsk-qwen-14b-finetune-v1-epoch4-awq.yaml -------------------------------------------------------------------------------- /cfgs/dpsk-qwen-14b-finetune-v1-epoch4-repkv-awq-kv4.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/imagination-research/aimo2/HEAD/cfgs/dpsk-qwen-14b-finetune-v1-epoch4-repkv-awq-kv4.yaml -------------------------------------------------------------------------------- /cfgs/dpsk-qwen-14b-finetune-v1-epoch4-repkv-awq.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/imagination-research/aimo2/HEAD/cfgs/dpsk-qwen-14b-finetune-v1-epoch4-repkv-awq.yaml -------------------------------------------------------------------------------- /cfgs/dpsk-qwen-14b-finetune-v1-epoch4.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/imagination-research/aimo2/HEAD/cfgs/dpsk-qwen-14b-finetune-v1-epoch4.yaml -------------------------------------------------------------------------------- /cfgs/dpsk-qwen-14b.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/imagination-research/aimo2/HEAD/cfgs/dpsk-qwen-14b.yaml -------------------------------------------------------------------------------- /data/aime_2025_30.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/imagination-research/aimo2/HEAD/data/aime_2025_30.csv -------------------------------------------------------------------------------- /data/reference.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/imagination-research/aimo2/HEAD/data/reference.csv -------------------------------------------------------------------------------- /data/stub.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/imagination-research/aimo2/HEAD/data/stub.csv -------------------------------------------------------------------------------- /directory_structure.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/imagination-research/aimo2/HEAD/directory_structure.txt -------------------------------------------------------------------------------- /entry_points.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/imagination-research/aimo2/HEAD/entry_points.md -------------------------------------------------------------------------------- /figs/dpo_result.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/imagination-research/aimo2/HEAD/figs/dpo_result.png -------------------------------------------------------------------------------- /figs/early_stop_analysis.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/imagination-research/aimo2/HEAD/figs/early_stop_analysis.jpg -------------------------------------------------------------------------------- /figs/inference_workflow.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/imagination-research/aimo2/HEAD/figs/inference_workflow.png -------------------------------------------------------------------------------- /figs/initialization_com.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/imagination-research/aimo2/HEAD/figs/initialization_com.png -------------------------------------------------------------------------------- /figs/output_speed.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/imagination-research/aimo2/HEAD/figs/output_speed.png -------------------------------------------------------------------------------- /figs/sft_accuracy.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/imagination-research/aimo2/HEAD/figs/sft_accuracy.png -------------------------------------------------------------------------------- /figs/sft_output_len.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/imagination-research/aimo2/HEAD/figs/sft_output_len.png -------------------------------------------------------------------------------- /figs/throughput_com.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/imagination-research/aimo2/HEAD/figs/throughput_com.png -------------------------------------------------------------------------------- /figs/token_length.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/imagination-research/aimo2/HEAD/figs/token_length.png -------------------------------------------------------------------------------- /imagination_aimo2/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /imagination_aimo2/aggregators.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/imagination-research/aimo2/HEAD/imagination_aimo2/aggregators.py -------------------------------------------------------------------------------- /imagination_aimo2/backup_submission.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/imagination-research/aimo2/HEAD/imagination_aimo2/backup_submission.ipynb -------------------------------------------------------------------------------- /imagination_aimo2/local_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/imagination-research/aimo2/HEAD/imagination_aimo2/local_eval.py -------------------------------------------------------------------------------- /imagination_aimo2/local_eval_kaggle.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/imagination-research/aimo2/HEAD/imagination_aimo2/local_eval_kaggle.py -------------------------------------------------------------------------------- /kaggle_evaluation/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/imagination-research/aimo2/HEAD/kaggle_evaluation/__init__.py -------------------------------------------------------------------------------- /kaggle_evaluation/aimo_2_gateway.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/imagination-research/aimo2/HEAD/kaggle_evaluation/aimo_2_gateway.py -------------------------------------------------------------------------------- /kaggle_evaluation/aimo_2_inference_server.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/imagination-research/aimo2/HEAD/kaggle_evaluation/aimo_2_inference_server.py -------------------------------------------------------------------------------- /kaggle_evaluation/core/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/imagination-research/aimo2/HEAD/kaggle_evaluation/core/__init__.py -------------------------------------------------------------------------------- /kaggle_evaluation/core/base_gateway.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/imagination-research/aimo2/HEAD/kaggle_evaluation/core/base_gateway.py -------------------------------------------------------------------------------- /kaggle_evaluation/core/generated/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/imagination-research/aimo2/HEAD/kaggle_evaluation/core/generated/__init__.py -------------------------------------------------------------------------------- /kaggle_evaluation/core/generated/kaggle_evaluation_pb2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/imagination-research/aimo2/HEAD/kaggle_evaluation/core/generated/kaggle_evaluation_pb2.py -------------------------------------------------------------------------------- /kaggle_evaluation/core/generated/kaggle_evaluation_pb2_grpc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/imagination-research/aimo2/HEAD/kaggle_evaluation/core/generated/kaggle_evaluation_pb2_grpc.py -------------------------------------------------------------------------------- /kaggle_evaluation/core/kaggle_evaluation.proto: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/imagination-research/aimo2/HEAD/kaggle_evaluation/core/kaggle_evaluation.proto -------------------------------------------------------------------------------- /kaggle_evaluation/core/relay.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/imagination-research/aimo2/HEAD/kaggle_evaluation/core/relay.py -------------------------------------------------------------------------------- /kaggle_evaluation/core/templates.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/imagination-research/aimo2/HEAD/kaggle_evaluation/core/templates.py -------------------------------------------------------------------------------- /kaggle_forum.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/imagination-research/aimo2/HEAD/kaggle_forum.md -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/imagination-research/aimo2/HEAD/pyproject.toml -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/imagination-research/aimo2/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/analyze_early_stop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/imagination-research/aimo2/HEAD/scripts/analyze_early_stop.py -------------------------------------------------------------------------------- /scripts/analyze_llm_vote.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/imagination-research/aimo2/HEAD/scripts/analyze_llm_vote.py -------------------------------------------------------------------------------- /scripts/deduplicate_sft_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/imagination-research/aimo2/HEAD/scripts/deduplicate_sft_data.py -------------------------------------------------------------------------------- /scripts/filter_dpo_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/imagination-research/aimo2/HEAD/scripts/filter_dpo_data.py -------------------------------------------------------------------------------- /scripts/plot_table_local.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/imagination-research/aimo2/HEAD/scripts/plot_table_local.py -------------------------------------------------------------------------------- /scripts/quant_awq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/imagination-research/aimo2/HEAD/scripts/quant_awq.py -------------------------------------------------------------------------------- /scripts/quant_rep_kv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/imagination-research/aimo2/HEAD/scripts/quant_rep_kv.py -------------------------------------------------------------------------------- /scripts/run_cfg.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/imagination-research/aimo2/HEAD/scripts/run_cfg.sh -------------------------------------------------------------------------------- /scripts/run_convert.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/imagination-research/aimo2/HEAD/scripts/run_convert.sh -------------------------------------------------------------------------------- /scripts/run_deduplicate.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/imagination-research/aimo2/HEAD/scripts/run_deduplicate.sh -------------------------------------------------------------------------------- /scripts/run_dpo.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/imagination-research/aimo2/HEAD/scripts/run_dpo.sh -------------------------------------------------------------------------------- /scripts/run_filter.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/imagination-research/aimo2/HEAD/scripts/run_filter.sh -------------------------------------------------------------------------------- /scripts/run_quant_awq.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/imagination-research/aimo2/HEAD/scripts/run_quant_awq.sh -------------------------------------------------------------------------------- /scripts/run_quant_rep_kv.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/imagination-research/aimo2/HEAD/scripts/run_quant_rep_kv.sh -------------------------------------------------------------------------------- /scripts/run_sft.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/imagination-research/aimo2/HEAD/scripts/run_sft.sh -------------------------------------------------------------------------------- /settings.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/imagination-research/aimo2/HEAD/settings.json --------------------------------------------------------------------------------