Pre- and post-activation processing support Grouped QVAFs for efficient training Plot the nodes and pruning unnecessary nodes Layer extension for more complex features and more ... A basic PennyLane ...

We welcome contributions to Trax! We welcome PRs with code for new models and layers as well as improvements to our code and documentation. We especially love notebooks that explain how models work ...