Sean Moriarity
Sean Moriarity
Related to discussion in #11 and would resolve https://github.com/elixir-nx/axon/issues/25 I have been digging in to input pipelines, specifically [tf.data](https://www.tensorflow.org/guide/data) and [torch.utils.data.DataLoader](https://pytorch.org/docs/stable/data.html#torch.utils.data.DataLoader). I'm more familiar with `tf.data`, and it's got a...
With the addition of RNNs in Axon, I'd like to start adding some examples of using them in practice. @t-rutten would you be open to adding some NLP related datasets...
Trying to align more with Flax's implementation which does not split keys but rather generates keys from layer meta
WIP: Need to add tests From our original implementation I added a few features, and there's still more I think we need to determine before merging: 1. `preprocess`/`postprocess` - I...
Requires https://github.com/elixir-nx/nx/issues/765
Requires https://github.com/elixir-nx/nx/issues/645
Right now large model compiler stacktraces are very difficult to interpret. It's almost impossible to tell where your issue is because the compiler offers no straightforward information in the stacktrace....