brett koonce

adamax/AMSGrad not working with differentiableReduce

2

all credit to @s1ddok for reporting this in https://github.com/tensorflow/swift-models/pull/394! something is failing (i think block-based reduces) with the adamax/AMSGrad optimizers: [edit: updated 2020-05-17 to match latest swift-models api changes, still...

imagenet demo not working out of the box

1

see also https://github.com/google/flax/issues/2403 flax 0.6 moved some things about, the resnet/vit demos will need to be updated: `python3 scenic/main.py --config=scenic/projects/baselines/configs/imagenet/imagenet_resnet_config.py --workdir=/code/resnet-v3` ``` 2022-08-19 01:19:21.743758: W tensorflow/stream_executor/platform/default/dso_loader.cc:64] Could not load dynamic...

rocm 5.7.1 + 7900 xtx + jax:latest docker image not working

11

hardware: 7900xtx ubuntu 22.04 lts rocm 5.7.1 (first version w/ official 7900xtx support) I am able to run the pytorch image, eg: ``` drun rocm/pytorch root@minerva:~# python3 Python 3.9.18 (main,...

### Description ``` docker pull rocm/jax Using default tag: latest latest: Pulling from rocm/jax Digest: sha256:1d0238004c3049426a6a91e64fd0ef9f19c63e2418ceb34a4b1fcbc93e4f510e Status: Image is up to date for rocm/jax:latest docker.io/rocm/jax:latest ``` > drun rocm/jax:latest >...

### Description @rahulbatra85 I managed to get jax to train resnet50 using the scenic library and the latest docker images, thanks to #18747. When I change my code from `float32`...

[Issue]: ROCm 6.1: ModuleNotFoundError: No module named 'yaml'

5

### Problem Description installed rocm 6.1, getting this error OS: NAME="Ubuntu" VERSION="22.04.4 LTS (Jammy Jellyfish)" CPU: model name : AMD Ryzen 9 3950X 16-Core Processor GPU: Name: AMD Ryzen 9...