search

elixir-nx / bumblebee

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

Reduce memory used by :preallocate_params #317

Closed jonatanklosko closed 9 months ago

jonatanklosko commented 9 months ago

See https://github.com/elixir-nx/bumblebee/issues/147#issuecomment-1862536942 and https://github.com/elixir-nx/nx/pull/1429.