Draft: Add `@distribute` decorator for parallelizing functions#123

Open

engintoklu wants to merge 10 commits intomasterfrom

feature/functional-distribute

Commits on Sep 12, 2025

Add `@distribute` decorator for parallelizing functions
engintoklu
committed

Commits on Sep 17, 2025

Redesign `on_device`, `on_cuda`, `on_aux_device`
engintoklu
committed

Commits on Sep 18, 2025

Add the `chunk_size` argument to the `distribute` decorator
engintoklu
committed

Commits on Sep 25, 2025

Commits on Sep 29, 2025

Some fixes & ability to mark non-tensor args with `on_device`
engintoklu
committed