Add axis option to poissonmultinomial loss

Genentech / gReLU

gReLU is a python library to train, interpret, and apply deep learning models to DNA sequences.

https://genentech.github.io/gReLU/

MIT License

228 stars 23 forks source link

Closed avantikalal closed 2 weeks ago

avantikalal commented 1 month ago

Add an option to apply poissonmultinomial loss across the task axis, not just the length axis.