class_docstr: Convert waveform to mel spectrogram.
class_docstr_tok: ["Convert","waveform","to","mel","spectrogram","."]
code: def __init__(
      self,
      sample_rate: int,
      n_fft: int,
      n_mels: int,
      fmin=0.0,
      fmax=8000,
  ):
      self.melfb = jax.device_put(
          librosa.filters.mel(
              sr=sample_rate,
              n_fft=n_fft,
              n_mels=n_mels,
              fmin=fmin,
              fmax=fmax,
          )
      )
      self.n_fft = n_fft
  
file_path: underthesea/underthesea/pipeline/say/viettts_/nat/dsp.py
imports: from functools import partial
  from typing import Optional
  import jax
  import jax.numpy
  import librosa
  from einops import rearrange
  from jax.numpy import ndarray
label_desc: Use image node_id 1 to create a new MelFilter object  with example: obj = MelFilter(sample_rate, n_fft, n_mels, fmin, fmax)