Donโ€™t abstract jax๐Ÿ˜”