Scicom-AI-Enterprise-Organization/titanet-vectors-fp16

TitaNet-L vectors output in FP16.

Python

0

3 commits

updated Jan 24, 2026

See the code

README

titanet-vectors-fp16

TitaNet-L vectors output in FP16.

How to

  1. Install,
pip3 install git+https://github.com/Scicom-AI-Enterprise-Organization/titanet-vectors-fp16
  1. Follow load.ipynb.

Small things that we learnt

  1. STFT always returned in complex64, https://github.com/Scicom-AI-Enterprise-Organization/titanet-vectors-fp16/blob/main/titanet_vectors/nemo_featurization.py#L464
  2. when you want to get the real value, it will always in fp32, https://github.com/Scicom-AI-Enterprise-Organization/titanet-vectors-fp16/blob/main/titanet_vectors/nemo_featurization.py#L472
  3. Maintain stats precision in fp32, but after that downcast to expected dtype before attention, https://github.com/Scicom-AI-Enterprise-Organization/titanet-vectors-fp16/blob/main/titanet_vectors/nemo/tdnn_attention.py#L319

Scicom-AI-Enterprise-Organization/titanet-vectors-fp16

TitaNet-L vectors output in FP16.

Python

0

3 commits

updated Jan 24, 2026

See the code

README

titanet-vectors-fp16

TitaNet-L vectors output in FP16.

How to

  1. Install,
pip3 install git+https://github.com/Scicom-AI-Enterprise-Organization/titanet-vectors-fp16
  1. Follow load.ipynb.

Small things that we learnt

  1. STFT always returned in complex64, https://github.com/Scicom-AI-Enterprise-Organization/titanet-vectors-fp16/blob/main/titanet_vectors/nemo_featurization.py#L464
  2. when you want to get the real value, it will always in fp32, https://github.com/Scicom-AI-Enterprise-Organization/titanet-vectors-fp16/blob/main/titanet_vectors/nemo_featurization.py#L472
  3. Maintain stats precision in fp32, but after that downcast to expected dtype before attention, https://github.com/Scicom-AI-Enterprise-Organization/titanet-vectors-fp16/blob/main/titanet_vectors/nemo/tdnn_attention.py#L319