Problem:
Find the jacobian of the vector valued function $\begin{pmatrix} f(x,y,z)\\ g(x,y,z) \\ h(x,y,z)\end{pmatrix}$ where $f(x,y,z) = \tan ^{-1}\left(x+y^2\right)$, $g(x,y,z) = y^2$, and $h(x,y,z) = \sqrt{\frac{z}{x}}$
from sympy import *
from sympy.abc import x, y, z

f = atan(x+y**2)
g = y**2
h = sqrt((z/x))
X = Matrix([f, g, h]) 
print(X.jacobian((x, y, z)))
