A handful of efficiency optimizations - #19

Merged

psheehan merged 8 commits into

mainfrom

efficiency_optimizations

Jun 21, 2026

Commits on Jun 18, 2026

torch.no_grad for evaluating ML models as it speeds up calculations.
psheehan
committed
b2d3cf1
Put source photon/ray emission into torch tensors on the appropriate device so they get an appropriate speedup on execution.
psheehan
committed
53c1dab
Remove double calculation of photon opacities; doesn't provide a huge speedup, but still unnecessary.
psheehan
committed
d51df5b

Commits on Jun 19, 2026

Commits on Jun 20, 2026

Make sure direction and direction frame access different memory now that they are coming from torch. Also remove spurious exp that somehow ended up in the direction.
psheehan
committed
13024c9