Convert Regularization Utilities to Higher Order Functions

kmheckel / spyx

Spyx: Spiking Neural Networks in JAX

https://spyx.readthedocs.io/en/latest/

MIT License

98 stars 11 forks source link

Convert Regularization Utilities to Higher Order Functions #2

Closed kmheckel closed 11 months ago

kmheckel commented 12 months ago

I think it's appropriate for Spyx to adopt a functional approach as much as possible, so using functions that return other functions (higher order functions) rather than classes would be fitting since it helps guide the user towards JIT compiling.

[x] Convert spyx.fn.silence_reg to a H.O.F.
[x] Convert spyx.fn.sparsity_reg to a H.O.F.

kmheckel commented 11 months ago

I've converted the two classes to functions which return functions.