search

pynapple-org / pynapple

PYthon Neural Analysis Package :pineapple:
https://pynapple-org.github.io/pynapple/
MIT License
243 stars 59 forks source link

Implementation of Optimized Spectrogram Computation Using JAX #294

Open rafael-mannarelli opened 1 month ago

rafael-mannarelli commented 1 month ago
  1. The spectrogram function computes the magnitude of the STFT in decibels using JAX's efficient computation capabilities. 2. The compute_spectrogram function handles data input from memory-mapped files, making it suitable for extensive datasets, and organizes the computation across different channels and batches. 3. The plot_spectrogram function provides a dynamic way to visualize the spectrogram data, supporting both full-range and specific time interval views.