I am developing a gemoetric framework for feature extraction and learning algorithm designs. More information can be found in this , including some pytorch demos.