Implement multivariate Bernstein polynomials as a JAX-based model function.
Implement multivariate Bernstein polynomials as a JAX-based model function.