├── .gitignore ├── CODE_OF_CONDUCT.md ├── CONTRIBUTING.md ├── LICENSE ├── README.md ├── data_generator ├── jspDatagen.py └── vrpDatagen.py ├── figs ├── expression_simplification.png ├── job_scheduling.png └── vehicle_routing.png └── src ├── .gitignore ├── Halide_search.py ├── arguments.py ├── jsp_nonNN_baselines.py ├── models ├── BaseModel.py ├── HalideModel.py ├── __init__.py ├── data_utils │ ├── Dag.py │ ├── Seq.py │ ├── Tree.py │ ├── __init__.py │ ├── data_utils.py │ ├── parser.py │ └── utils.py ├── jspModel.py ├── model_utils │ ├── __init__.py │ ├── logger.py │ └── supervisor.py ├── modules │ ├── HalideInputEncoder.py │ ├── __init__.py │ ├── jspInputEncoder.py │ ├── mlp.py │ └── vrpInputEncoder.py ├── rewriter │ ├── HalideRewriter.py │ ├── __init__.py │ ├── jspRewriter.py │ └── vrpRewriter.py └── vrpModel.py ├── run_Halide.py ├── run_jsp.py └── run_vrp.py /.gitignore: -------------------------------------------------------------------------------- 1 | *.swp 2 | *.json 3 | *.csv 4 | -------------------------------------------------------------------------------- /CODE_OF_CONDUCT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/neural-rewriter/HEAD/CODE_OF_CONDUCT.md -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/neural-rewriter/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/neural-rewriter/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/neural-rewriter/HEAD/README.md -------------------------------------------------------------------------------- /data_generator/jspDatagen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/neural-rewriter/HEAD/data_generator/jspDatagen.py -------------------------------------------------------------------------------- /data_generator/vrpDatagen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/neural-rewriter/HEAD/data_generator/vrpDatagen.py -------------------------------------------------------------------------------- /figs/expression_simplification.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/neural-rewriter/HEAD/figs/expression_simplification.png -------------------------------------------------------------------------------- /figs/job_scheduling.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/neural-rewriter/HEAD/figs/job_scheduling.png -------------------------------------------------------------------------------- /figs/vehicle_routing.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/neural-rewriter/HEAD/figs/vehicle_routing.png -------------------------------------------------------------------------------- /src/.gitignore: -------------------------------------------------------------------------------- 1 | __pycache__ 2 | -------------------------------------------------------------------------------- /src/Halide_search.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/neural-rewriter/HEAD/src/Halide_search.py -------------------------------------------------------------------------------- /src/arguments.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/neural-rewriter/HEAD/src/arguments.py -------------------------------------------------------------------------------- /src/jsp_nonNN_baselines.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/neural-rewriter/HEAD/src/jsp_nonNN_baselines.py -------------------------------------------------------------------------------- /src/models/BaseModel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/neural-rewriter/HEAD/src/models/BaseModel.py -------------------------------------------------------------------------------- /src/models/HalideModel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/neural-rewriter/HEAD/src/models/HalideModel.py -------------------------------------------------------------------------------- /src/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/models/data_utils/Dag.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/neural-rewriter/HEAD/src/models/data_utils/Dag.py -------------------------------------------------------------------------------- /src/models/data_utils/Seq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/neural-rewriter/HEAD/src/models/data_utils/Seq.py -------------------------------------------------------------------------------- /src/models/data_utils/Tree.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/neural-rewriter/HEAD/src/models/data_utils/Tree.py -------------------------------------------------------------------------------- /src/models/data_utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/models/data_utils/data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/neural-rewriter/HEAD/src/models/data_utils/data_utils.py -------------------------------------------------------------------------------- /src/models/data_utils/parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/neural-rewriter/HEAD/src/models/data_utils/parser.py -------------------------------------------------------------------------------- /src/models/data_utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/neural-rewriter/HEAD/src/models/data_utils/utils.py -------------------------------------------------------------------------------- /src/models/jspModel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/neural-rewriter/HEAD/src/models/jspModel.py -------------------------------------------------------------------------------- /src/models/model_utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/neural-rewriter/HEAD/src/models/model_utils/__init__.py -------------------------------------------------------------------------------- /src/models/model_utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/neural-rewriter/HEAD/src/models/model_utils/logger.py -------------------------------------------------------------------------------- /src/models/model_utils/supervisor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/neural-rewriter/HEAD/src/models/model_utils/supervisor.py -------------------------------------------------------------------------------- /src/models/modules/HalideInputEncoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/neural-rewriter/HEAD/src/models/modules/HalideInputEncoder.py -------------------------------------------------------------------------------- /src/models/modules/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/models/modules/jspInputEncoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/neural-rewriter/HEAD/src/models/modules/jspInputEncoder.py -------------------------------------------------------------------------------- /src/models/modules/mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/neural-rewriter/HEAD/src/models/modules/mlp.py -------------------------------------------------------------------------------- /src/models/modules/vrpInputEncoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/neural-rewriter/HEAD/src/models/modules/vrpInputEncoder.py -------------------------------------------------------------------------------- /src/models/rewriter/HalideRewriter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/neural-rewriter/HEAD/src/models/rewriter/HalideRewriter.py -------------------------------------------------------------------------------- /src/models/rewriter/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/neural-rewriter/HEAD/src/models/rewriter/__init__.py -------------------------------------------------------------------------------- /src/models/rewriter/jspRewriter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/neural-rewriter/HEAD/src/models/rewriter/jspRewriter.py -------------------------------------------------------------------------------- /src/models/rewriter/vrpRewriter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/neural-rewriter/HEAD/src/models/rewriter/vrpRewriter.py -------------------------------------------------------------------------------- /src/models/vrpModel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/neural-rewriter/HEAD/src/models/vrpModel.py -------------------------------------------------------------------------------- /src/run_Halide.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/neural-rewriter/HEAD/src/run_Halide.py -------------------------------------------------------------------------------- /src/run_jsp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/neural-rewriter/HEAD/src/run_jsp.py -------------------------------------------------------------------------------- /src/run_vrp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/neural-rewriter/HEAD/src/run_vrp.py --------------------------------------------------------------------------------