search

elixir-nx / bumblebee

Pre-trained Neural Network models in Axon (+ 🤗 Models integration)
Apache License 2.0
1.33k stars 96 forks source link

Vectorize logits processors and add more tests #241

Closed jonatanklosko closed 1 year ago

jonatanklosko commented 1 year ago

Should make the implementations easier to follow :)