Cosmic ray propagation with high dimensional finite element method