jax_bidirectional_tree_rake_compress
