JAX is a good example of how to use the same method of writing a code in a different way than the one you used in the previous post and it is a great way to use it in a different JAX environment and to use it in a different environment in a different context Ponmi is a good one for that but I don’t think it’s a good
