A simple Jax implementation of influence functions.
Primary LanguagePythonApache License 2.0Apache-2.0