したこと Jaxでvggモデル実装 from functools import partial from typing import Callable, Optional, Sequence, Tuple, List, Union import jax.numpy as jnp from flax import linen as nn from .common import ConvBlock, ModuleDef, Sequential ModuleD…