Performance Improvement

The performance isn't good when input is big. For example, a matrix of 100 × 50000 To benchmark current state: ```julia using PkgBenchmark, ContinuousNormalizingFlows benchmarkpkg(ContinuousNormalizingFlows) ```