jglaser

Yup, starting the network with a `Dense` layer works

Please let me know if you need me to test any code on AMD.

> A lot of rocm_smi function require the mutex. In the [unit test](https://github.com/RadeonOpenCompute/rocm_smi_lib/blob/master/tests/rocm_smi_test/functional/mutual_exclusion.cc#L264), you can find most of them. > > I tried to reproduce this issue in my machine...

is this still being worked on?

awesome, thanks!

It would be good if numpy data type extensions à la https://github.com/jax-ml/ml_dtypes/tree/main were supported, which includes bfloat16, fp8 etc.