pytc.ansatz.gto.eval_ao¶

pytc.ansatz.gto.eval_ao(mol_gto, pos, deriv=0)[source]¶

JAX-compatible eval_ao. pos: (batch…, 3)

Parameters:

mol_gto (flax.struct.dataclass)
pos (jax.Array)