[JAX] Move parallel encoder tests to L0 distributed test set. (#1356)

[JAX] Move parallel encoder tests to L0 distributed test set. (#1356) #755