change a lot

This commit is contained in:
wls2002
2023-07-17 19:59:46 +08:00
parent f4763ebcea
commit 40cf0b6fbe
8 changed files with 248 additions and 18 deletions

View File

@@ -1,6 +1,5 @@
from typing import Dict, Tuple, Type
import numpy as np
import jax
from jax import Array, numpy as jnp, vmap