Henry Ndubuaku

icon indicating an email [email protected]

icon company Cactus Compute icon location London icon location EE + Maths + Computing + Cognition

Results 2 comments of Henry Ndubuaku

Jax version

Thanks for this, it seems to work on my end. ```py import jax, nanodl print(f"JAX version: {jax.__version__}") print(f"JAX version: {nanodl.__version__}") ``` ``` JAX version: 0.4.25 JAX version: 1.2.0.dev1 ``` Does...

Gradient synchronization in data-parallel trainers

Thanks for noticing this! It's often challenging to test these portions due to the unavailability of a personal multi-GPU setup for development. However, I will be accessing 2 GPUs around...