Apoorv Gupta

icon indicating an email [email protected]

icon company Amazon Web Services icon location Cupertino, CA icon location Software Engineer ll at Amazon Web Services Neuron

Results 8 comments of Apoorv Gupta

Smddp MNIST example update

> @apoorvtintin Thanks for the PR! Code changes look good to me! Would you help update the notebook too? There's some outdated info e.g. > > 1. "This notebook example...

Neuron support in Axlearn

Thanks for all the reviews, I fixed most of the comments on the PR.

Neuron support in Axlearn

Not needed, closing this

Fallback to standard mesh on neuron backend for incompatible multi-granule meshes

Rebased the PR

Fallback to standard mesh on neuron backend for incompatible multi-granule meshes

Yes, I would like to move ahead with this, let me create a rebased PR.

Save O_PROJ on Fuji 70B-v2 for TRN2

Rebased the PR to fix failing CI

Save O_PROJ on Fuji 70B-v2 for TRN2

I see the CI fails for test `TestEvaluateFromFile.test_evaluate_from_eval_set` with error `#22 453.1 axlearn/open_api/common.py:440: KeyError` This is unrelated to the changes in this PR, I already rebased the PR to 12th...

Save O_PROJ on Fuji 70B-v2 for TRN2

Rebased to disable flaky test