Closed OliverFM closed 1 year ago
Added a working version of the bmm_jvp. This is the main blocker that remains for training an MLP network.
bmm_jvp
The implementation is correct, I think, but could use a lot more testing, and could almost certainly be made much cleaner, and, of course, faster.
Added a working version of the
bmm_jvp. This is the main blocker that remains for training an MLP network.The implementation is correct, I think, but could use a lot more testing, and could almost certainly be made much cleaner, and, of course, faster.