nmf - find ways to speed up model

cellarium-ai / cellarium-ml

Distributed single-cell data analysis.

BSD 3-Clause "New" or "Revised" License

22 stars 3 forks source link

Open sjfleming opened 2 months ago

sjfleming commented 2 months ago

Look into torch.compile again

Figure out some way to make the model as fast as we reasonably can.

This is not currently a blocker, but if we don't pursue it, somebody else will come out with a faster version of this.

sjfleming commented 2 months ago

Nested for-loops in the non-negative least squares solver update are the target.

sjfleming commented 1 day ago

See Nimish and Ebtisam's example here