Recent Releases of AutoPDEx: An Automized Partial Differential Equation solver based on JAX