d6bcc56001
* Added confusion metrics -- still using TF ops * Fixed structure + tests pass for TF (still need to port to multi-backend) * Got rid of most tf deps, still a few more to go * Full removal of TF. Tests pass for both Jax and TF * Full removal of TF. Tests pass for both Jax and TF * Formatting * Formatting * Review comments * More review comments + formatting |
||
|---|---|---|
| .. | ||
| __init__.py | ||
| math.py | ||
| nn.py | ||
| numpy.py | ||
| random.py | ||
| trainer.py | ||