David Tweedle

icon indicating an email [email protected]

icon company University of the West Indies, St. Augustine Campus icon location St. Augustine, Trinidad icon location Lecturer in the Department of Mathematics and Statistics, Faculty of Science and Technology, University of the West Indies, St. Augustine Campus.

Results 2 issues of David Tweedle

Jit switch for ogbg

2 comment

I added some code to the ogbg workload according to the migration guide helpfully provided by Ahmed. Unfortunately, there seems to be some bugs that I still hope to fix.

Cifar dataloader does not work

2 comment

The cifar dataloader no longer works properly with jax algorithms using jax.jit. I did not test to see if pytorch algorithms still work with cifar. ## Description When running jax_nadamw_full_budget.py...