INNER CODE UNIT · Python

test_flax

HenryNdubuaku/nanodl · nanodl/__init__.py:315

def test_flax(flax):
    model = flax.linen.Dense(features=10)


def test_jax(jax):
    arr = jax.numpy.array([1, 2, 3])
    result = jax.numpy.sum(arr)


def test_optax(optax):
    optimizer = optax.sgd(learning_rate=0.1)


def test_einops(einops):
    arr = einops.rearrange([1, 2, 3], "a b c -> b a c")


def main():

View source record →