Code

Non contiguous transpose axis#28

Code

Merged

ASKabalan merged 4 commits intomainDifferentiableUniverseInitiative/jaxDecomp:mainfrom non-contiguous-transpose-axisDifferentiableUniverseInitiative/jaxDecomp:non-contiguous-transpose-axisCopy head branch name to clipboardOct 23, 2024

+737-579

Commits

Commits on Aug 20, 2024

• implement non-contiguous distributed transpose

  ASKabalan

  committed
  3d871a3

• implement non-contiguous distributed FFTs

  ASKabalan

  committed
  989243c

• add tests for non contiguous FFTs and transposes

  ASKabalan

  committed
  d921300

Commits on Oct 6, 2024

• merge with main

  ASKabalan

  committed
  bfec8fb