christinahedges/scatterbrain

GPU version of tessbackdrop

scatterbrain

scatterbrain is our GPU hack for processing TESS images, see tess-backdrop for our current tool.

TODO

• Make starscene cupy
• Add files on Zenodo so we can load them if they aren't local
• Add docs to asteroid functions
• Add some tutorials for different use cases
• Come up with some diagnostics for backdrops/starscenes
• Check starscenes for multiple sectors
• Run on supercomputer
• Investigate using GPUs to run on super computer
• Increase the asteroid masking depth to mask fainter asteroids?
• Make asteroid_mask supersampled in time for fast, bright asteroids...?

NOTES:

• Consider swapping third order polynomial in StarScene to a low order spline for nicer priors and better variability removal?
  • Did this, now has the option. Overall, seems better to just use a polynomial.

scatterbrain's Issues

Strided arrays

@dmargala has pointed out that we're not actually using the batched version of the cholesky solve, and we can benefit from strided arrays to get that to work.

This is his snippet to get this to work

assert a2d.shape == (n, n)
a3d = cp.lib.stride_tricks.as_strided(a2d, shape=(k, n, n), strides=(0, n*a2d.itemsize, a2d.itemsize))
assert a3d.shape == (k, n, n)