├── .gitignore ├── LICENSE ├── README.md ├── assets ├── figure1.jpg ├── maze_plot_train_hmcts_00.jpg └── maze_scalability.png ├── environment.yml ├── requirements.txt ├── scripts ├── Maze_inference.sh ├── Maze_train.sh ├── Sudoku_inference.sh └── Sudoku_train.sh ├── setup.py └── tscend_src ├── __init__.py ├── configs └── config_base.yaml ├── data ├── MazeDataset_generator.ipynb ├── __init__.py ├── data_maze.py ├── dataset.py ├── gen_planning_dataset.py ├── planning_dataset.py ├── reasoning_dataset.py └── sat_dataset.py ├── filepath.py ├── inference ├── inference_Maze.py └── inference_Sudoku.py ├── model ├── __init__.py ├── diffusion_lib │ ├── denoising_diffusion_pytorch_1d.py │ ├── nlm.py │ ├── nlm_utils.py │ └── transformer.py └── models.py ├── train ├── __init__.py └── train.py └── utils ├── __init__.py ├── utils.py ├── utils_maze_continuous.py └── utils_sudoku_continuous.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AI4Science-WestlakeU/t_scend/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AI4Science-WestlakeU/t_scend/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AI4Science-WestlakeU/t_scend/HEAD/README.md -------------------------------------------------------------------------------- /assets/figure1.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AI4Science-WestlakeU/t_scend/HEAD/assets/figure1.jpg -------------------------------------------------------------------------------- /assets/maze_plot_train_hmcts_00.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AI4Science-WestlakeU/t_scend/HEAD/assets/maze_plot_train_hmcts_00.jpg -------------------------------------------------------------------------------- /assets/maze_scalability.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AI4Science-WestlakeU/t_scend/HEAD/assets/maze_scalability.png -------------------------------------------------------------------------------- /environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AI4Science-WestlakeU/t_scend/HEAD/environment.yml -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AI4Science-WestlakeU/t_scend/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/Maze_inference.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AI4Science-WestlakeU/t_scend/HEAD/scripts/Maze_inference.sh -------------------------------------------------------------------------------- /scripts/Maze_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AI4Science-WestlakeU/t_scend/HEAD/scripts/Maze_train.sh -------------------------------------------------------------------------------- /scripts/Sudoku_inference.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AI4Science-WestlakeU/t_scend/HEAD/scripts/Sudoku_inference.sh -------------------------------------------------------------------------------- /scripts/Sudoku_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AI4Science-WestlakeU/t_scend/HEAD/scripts/Sudoku_train.sh -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AI4Science-WestlakeU/t_scend/HEAD/setup.py -------------------------------------------------------------------------------- /tscend_src/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tscend_src/configs/config_base.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AI4Science-WestlakeU/t_scend/HEAD/tscend_src/configs/config_base.yaml -------------------------------------------------------------------------------- /tscend_src/data/MazeDataset_generator.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AI4Science-WestlakeU/t_scend/HEAD/tscend_src/data/MazeDataset_generator.ipynb -------------------------------------------------------------------------------- /tscend_src/data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tscend_src/data/data_maze.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AI4Science-WestlakeU/t_scend/HEAD/tscend_src/data/data_maze.py -------------------------------------------------------------------------------- /tscend_src/data/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AI4Science-WestlakeU/t_scend/HEAD/tscend_src/data/dataset.py -------------------------------------------------------------------------------- /tscend_src/data/gen_planning_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AI4Science-WestlakeU/t_scend/HEAD/tscend_src/data/gen_planning_dataset.py -------------------------------------------------------------------------------- /tscend_src/data/planning_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AI4Science-WestlakeU/t_scend/HEAD/tscend_src/data/planning_dataset.py -------------------------------------------------------------------------------- /tscend_src/data/reasoning_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AI4Science-WestlakeU/t_scend/HEAD/tscend_src/data/reasoning_dataset.py -------------------------------------------------------------------------------- /tscend_src/data/sat_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AI4Science-WestlakeU/t_scend/HEAD/tscend_src/data/sat_dataset.py -------------------------------------------------------------------------------- /tscend_src/filepath.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AI4Science-WestlakeU/t_scend/HEAD/tscend_src/filepath.py -------------------------------------------------------------------------------- /tscend_src/inference/inference_Maze.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AI4Science-WestlakeU/t_scend/HEAD/tscend_src/inference/inference_Maze.py -------------------------------------------------------------------------------- /tscend_src/inference/inference_Sudoku.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AI4Science-WestlakeU/t_scend/HEAD/tscend_src/inference/inference_Sudoku.py -------------------------------------------------------------------------------- /tscend_src/model/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tscend_src/model/diffusion_lib/denoising_diffusion_pytorch_1d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AI4Science-WestlakeU/t_scend/HEAD/tscend_src/model/diffusion_lib/denoising_diffusion_pytorch_1d.py -------------------------------------------------------------------------------- /tscend_src/model/diffusion_lib/nlm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AI4Science-WestlakeU/t_scend/HEAD/tscend_src/model/diffusion_lib/nlm.py -------------------------------------------------------------------------------- /tscend_src/model/diffusion_lib/nlm_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AI4Science-WestlakeU/t_scend/HEAD/tscend_src/model/diffusion_lib/nlm_utils.py -------------------------------------------------------------------------------- /tscend_src/model/diffusion_lib/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AI4Science-WestlakeU/t_scend/HEAD/tscend_src/model/diffusion_lib/transformer.py -------------------------------------------------------------------------------- /tscend_src/model/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AI4Science-WestlakeU/t_scend/HEAD/tscend_src/model/models.py -------------------------------------------------------------------------------- /tscend_src/train/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tscend_src/train/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AI4Science-WestlakeU/t_scend/HEAD/tscend_src/train/train.py -------------------------------------------------------------------------------- /tscend_src/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tscend_src/utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AI4Science-WestlakeU/t_scend/HEAD/tscend_src/utils/utils.py -------------------------------------------------------------------------------- /tscend_src/utils/utils_maze_continuous.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AI4Science-WestlakeU/t_scend/HEAD/tscend_src/utils/utils_maze_continuous.py -------------------------------------------------------------------------------- /tscend_src/utils/utils_sudoku_continuous.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AI4Science-WestlakeU/t_scend/HEAD/tscend_src/utils/utils_sudoku_continuous.py --------------------------------------------------------------------------------