Distributed data parallel accelerator agnostic model#1838

Open

ajayvohra2005 wants to merge 18 commits intoMegvii-BaseDetection:mainfrom

ajayvohra2005:main

Commits on Oct 7, 2024

dev commit

ajayvohra2005
authored and
Ubuntu
committed
dev commit

ajayvohra2005
authored and
Ubuntu
committed
add xla dockerfile

ajayvohra2005
authored and
Ubuntu
committed
parse dtype

ajayvohra2005
authored and
Ubuntu
committed

Commits on Oct 8, 2024

update xla code

ajayvohra2005
authored and
Ubuntu
committed

Commits on Oct 9, 2024

dev commit

ajayvohra2005
authored and
Ubuntu
committed
first working version XLA on GPU

ajayvohra2005
authored and
Ubuntu
committed
distributed xla GPU fix

ajayvohra2005
authored and
Ubuntu
committed

Commits on Oct 13, 2024

fix xla gather

ajayvohra2005
authored and
Ubuntu
committed

Commits on Oct 17, 2024

update for xla

ajayvohra2005
authored and
Ubuntu
committed
update for neuron

ajayvohra2005
authored and
Ubuntu
committed
updates for neuron

ajayvohra2005
authored and
Ubuntu
committed

Commits on Oct 19, 2024

update xla code

ajayvohra2005
authored and
Ubuntu
committed
update scripts

ajayvohra2005
authored and
Ubuntu
committed

Commits on Jan 3, 2025

update code

ajayvohra2005
authored and
Ubuntu
committed

Commits on May 10, 2025

update scripts

ajayvohra2005
authored and
Ubuntu
committed
merge upstream

ajayvohra2005
authored and
Ubuntu
committed
update Dockerfile

ajayvohra2005
authored and
Ubuntu
committed