C. Antonio Sánchez

RFC: Tensor Indexed Updates

4

This is to allow numpy-like (or rather JAX-like) indexed tensor modifications like slice assignment.

DRAFT: Add custom variable updater.

2

Allows customization for how variables are updated by the optimizer. The base optimizer simply defers to the update handler to do the update, allowing full customization. Can replace the existing...

Allow per-variable optimizer, add DispatchOptimizer.

9

- Adds a property `variable.optimizer` that defaults to `None` - Adds a `DispatchOptimizer` that scans the list of trainable variables during build, collects all unique per-variable optimizers, then dispatches the...

eigen: Fix Eigen version comparison.

3

# Description The existing Eigen threadpool test uses an incorrect version comparison, which breaks with any new development branch or released version Eigen. Setting the dependency to anything newer than...