axon

axon copied to clipboard

WIP: The goal is to replace `*_unroll` entirely with this more general `scan` and to also use it to add both `Axon.scan` and `Axon.bidirectional` combinators

seanmor5

Bidirectional Wrapper Implementation (Feature Request)

6

Hello, @seanmor5 I had put up a bidirectional example in the Nx Slack channel. Currently planning out the implementation and looking at the use cases for it as follows: Tf/Keras...

fantypants

Remove ignore_batch logic in shape layers

1

It is just confusing, and I don't like the implicit behavior. It is easy enough for Axon to correctly do the reshape logic even with `nil` dimensions

seanmor5

Updated the humans_or_horses livebook

1

Added a really simple Exploratory Analysis section and configuration module. All hanging functions were put inside modules. The livebook can run to completion on my CPU.

meanderingstream

Converting ONNX file into Axon

1

The notebook presents an approach to using axon_onnx to parse an ONNX model into an Axon model. The notebook could be moved to axon_onnx, however, I think it belongs with...

meanderingstream

Make Axon.Updates state easier to query/work with

1

With the new container API, we can support structs. Right now the Axon.Updates API uses a tuple to represent optimizer state. It's not a bad, but I think it's better...

seanmor5

Text Classification LSTM Livebook & Example Integration

2

Hello, This is the initial implementation of the LSTM NER Text Classification. I've worked on the live-book & example for the model.

fantypants

We can implement table and mermaid displays in a separate Axon.Display module: - [x] `Axon.Display.as_table` - [ ] `Axon.Display.as_mermaid` And potentially many others :)

seanmor5

Start weight sharing API

1

Resolves #169 Still a few API considerations/questions: 1. Should we offer an explicit `:parameters` option on layer creation? 2. How should we handle partial parameter sharing, e.g. if I just...

seanmor5

And some minor changes

msluszniak