docs: public Pair<Gradient,INDArray> calculateGradients(@NonNull INDArray features, @NonNull INDArray label,
                                                        INDArray fMask, INDArray labelMask) {
          try{
              return calculateGradientsHelper(features, label, fMask, labelMask);
          } catch (OutOfMemoryError e){
              CrashReportingUtil.writeMemoryCrashDump(this, e);
              throw e;
          }
      }