asmith26/pandas_toolkit

Move loss functions to jax_toolkit

asmith26 opened this issue · 0 comments

asmith26 commented
  • Update jax_toolkit repo (and possible pandas_toolkit template) to more mirror one another.
  • Update doc (may need to rethink auto generation bits and pieces.
    • Maybe: Add visualisations of loss functions (see e.g. the bottom of this notebook)