add axis layer support
Merge branch 'ax_layer'