Tfc

Making changes to work with JAX 0.3.15.

**Full Changelog**: https://github.com/leakec/tfc/compare/tfc-v0.1.8...tfc-v0.1.9

* Updates so everything works with JAX 0.3.10.

**Full Changelog**: https://github.com/leakec/tfc/compare/tfc-v0.1.7...tfc-v0.1.8

* Minor changes to keep up with newest JAX update.

**Full Changelog**: https://github.com/leakec/tfc/compare/tfc-v0.1.6...tfc-v0.1.7

* Now offering `pe` and `pejit` that can make an argument constant. This will cache the argument and all functions that only depend on that argument as constants. These will be treated as compile time constants when using `pejit`
* This can speed up code noticeably but doesn't always. On `test_ODE` there was a speed gain of ~50x, but on `test_PDE` there was no noticeable difference.
* `constant_arg_nums` has been added to `LS`, `LsClass`, `NLLS`, and `NllsClass`. If this argument is used, then it will utilize `pejit` rather than `jit`.
* See 4 for more details.

TFC version 0.1.4.
* Major change is this implements 2.

Version 0.1.3 drops support for Python 3.6 and adds it for Python 3.10.