search

elixir-nx / nx

Multi-dimensional arrays (tensors) and numerical definitions for Elixir
2.55k stars 190 forks source link

fix: stop_grad for logsumexp #1470

Closed polvalente closed 3 months ago