Sean Moriarity
Sean Moriarity
@jnnks I would welcome a PR for resize! Let me know if you need any help. The Axon resize should have all of the pieces you need
You can just wrap an `Nx.sum` in an Axon layer, but you'll need to ensure you get axes using the Axon `constant!` helper. You won't be able to run the...
This happens from really expensive compilations. What specifically are you doing? Maybe we can find a way to simplify the executable you're trying to compile
An example would be if you're training an RNN, and unrolling the entire sequence then it will result in this error
I think we should update Axon to better support LoRA, I have a draft in place right now but I have to revisit it to make it work as I...
The multimodal projector should just be a FFN between the image and LLM, I can take a look at this
I actually think this is something we can do with the model state struct, since we can store metadata we can also tell when 2 parameters are tied. It's just...
Hey, this usually occurs because erlang-dev is packaged separately and is not installed: `brew install erlang-dev` Might resolve the issue
You'll need to add Erlangs include path to your system wide include directory; however, I'm in the process of refactoring this library, so it might be better to wait a...
For now to fix you can pin Axon to 0.6.1 while I work on fixing this