symmetric_nfbm_jax