search

LouisDesdoigts / zodiax

Object-oriented Jax framework extending Equinox for scientific programming
https://louisdesdoigts.github.io/zodiax/
BSD 3-Clause "New" or "Revised" License
11 stars 1 forks source link

Add default filter to `filter_grad` #14

Closed LouisDesdoigts closed 1 year ago

LouisDesdoigts commented 1 year ago

Default to filtering out strings to solve the string static_field() problem. Double check how this interacts with optax.

LouisDesdoigts commented 1 year ago

Default filters can be accessed through the equinox versions of the filter_functions.