add JUST wrapper

[lixinqi]

No description provided.

[github-actions]

Speed stats:

GPU Name: GeForce GTX 1080 

PyTorch resnet50 time: 141.9ms (= 7096.0ms / 50, input_shape=[16, 3, 224, 224], backward is enabled)
OneFlow resnet50 time: 126.7ms (= 6336.9ms / 50, input_shape=[16, 3, 224, 224], backward is enabled)
Relative speed: 1.12 (= 141.9ms / 126.7ms)

PyTorch resnet50 time: 85.1ms (= 4253.8ms / 50, input_shape=[8, 3, 224, 224], backward is enabled)
OneFlow resnet50 time: 74.2ms (= 3711.0ms / 50, input_shape=[8, 3, 224, 224], backward is enabled)
Relative speed: 1.15 (= 85.1ms / 74.2ms)

PyTorch resnet50 time: 58.0ms (= 2897.8ms / 50, input_shape=[4, 3, 224, 224], backward is enabled)
OneFlow resnet50 time: 47.6ms (= 2380.3ms / 50, input_shape=[4, 3, 224, 224], backward is enabled)
Relative speed: 1.22 (= 58.0ms / 47.6ms)

PyTorch resnet50 time: 49.6ms (= 2478.7ms / 50, input_shape=[2, 3, 224, 224], backward is enabled)
OneFlow resnet50 time: 41.9ms (= 2094.7ms / 50, input_shape=[2, 3, 224, 224], backward is enabled)
Relative speed: 1.18 (= 49.6ms / 41.9ms)

PyTorch resnet50 time: 43.9ms (= 2192.5ms / 50, input_shape=[1, 3, 224, 224], backward is enabled)
OneFlow resnet50 time: 41.6ms (= 2081.5ms / 50, input_shape=[1, 3, 224, 224], backward is enabled)
Relative speed: 1.05 (= 43.9ms / 41.6ms)

[github-actions]

Speed stats:

GPU Name: GeForce GTX 1080 

PyTorch resnet50 time: 140.7ms (= 7037.3ms / 50, input_shape=[16, 3, 224, 224], backward is enabled)
OneFlow resnet50 time: 126.8ms (= 6339.2ms / 50, input_shape=[16, 3, 224, 224], backward is enabled)
Relative speed: 1.11 (= 140.7ms / 126.8ms)

PyTorch resnet50 time: 84.9ms (= 4243.6ms / 50, input_shape=[8, 3, 224, 224], backward is enabled)
OneFlow resnet50 time: 74.0ms (= 3701.3ms / 50, input_shape=[8, 3, 224, 224], backward is enabled)
Relative speed: 1.15 (= 84.9ms / 74.0ms)

PyTorch resnet50 time: 58.3ms (= 2915.9ms / 50, input_shape=[4, 3, 224, 224], backward is enabled)
OneFlow resnet50 time: 49.3ms (= 2463.6ms / 50, input_shape=[4, 3, 224, 224], backward is enabled)
Relative speed: 1.18 (= 58.3ms / 49.3ms)

PyTorch resnet50 time: 47.1ms (= 2355.3ms / 50, input_shape=[2, 3, 224, 224], backward is enabled)
OneFlow resnet50 time: 45.9ms (= 2293.0ms / 50, input_shape=[2, 3, 224, 224], backward is enabled)
Relative speed: 1.03 (= 47.1ms / 45.9ms)

PyTorch resnet50 time: 40.8ms (= 2037.7ms / 50, input_shape=[1, 3, 224, 224], backward is enabled)
OneFlow resnet50 time: 45.3ms (= 2265.5ms / 50, input_shape=[1, 3, 224, 224], backward is enabled)
Relative speed: 0.90 (= 40.8ms / 45.3ms)