RES-2372: Add profiler and GPU Optimizations to dendrite experiments

numenta / nupic.research

Experimental algorithms. Unsupported.

https://nupicresearch.readthedocs.io

GNU Affero General Public License v3.0

104 stars 60 forks source link

Closed lscheinkman closed 3 years ago

lscheinkman commented 3 years ago

This PR adds a mixin enabling the newest torch.profiler (requires torch>=1.8.1) and a few optimizations. Depends on https://github.com/numenta/nupic.torch/pull/81