perf: benchmarking our models against Jax (Flax) #34

Job	Run time
downgrade (conv, default, true)	18m 10s
downgrade (dense, default, true)	1h 46m 0s
test (1.10, ubuntu-latest, conv, default, true)	19m 40s
downgrade (batch_norm, default, true)	19m 49s
test (1.10, ubuntu-latest, dense, default, true)	1h 46m 46s
downgrade (group_norm, default, true)	1h 11m 20s
test (1.10, ubuntu-latest, batch_norm, default, true)	20m 41s
downgrade (instance_norm, default, true)	16m 28s
test (1.10, ubuntu-latest, group_norm, default, true)	1h 14m 41s
downgrade (layer_norm, default, true)	1h 15m 47s
test (1.10, ubuntu-latest, instance_norm, default, true)	17m 2s
downgrade (other_ops, default, true)	35m 28s
test (1.10, ubuntu-latest, layer_norm, default, true)	1h 13m 58s
test (1.10, ubuntu-latest, other_ops, default, true)	35m 12s
downgrade (batched_ops, default, true)	22m 55s
test (1.10, ubuntu-latest, batched_ops, default, true)	21m 30s
downgrade (others, default, true)	13m 1s
test (1.10, ubuntu-latest, others, default, true)	13m 35s
test (ubuntu-latest, dense, blis, 1.10, true)	1h 47m 43s
test (ubuntu-latest, dense, mkl, 1.10, true)	1h 46m 52s
test (ubuntu-latest, dense, default, 1.10, false)	44m 9s
test (ubuntu-latest, batched_ops, default, 1.10, false)	26m 11s
test (ubuntu-latest, other_ops, default, 1.10, false)	38m 32s
test (macos-latest, dense, appleaccelerate, 1.10, true)	1h 9m 13s
test (macos-latest, all, default, 1.10, true)	6h 0m 3s
test (windows-latest, all, default, 1.10, true)	6h 0m 0s
	1d 7h 14m 46s

Provide feedback