| File | Mode | Size |
|---|---|---|
| jax2tf | ||
| README.md | -rw-r--r-- | 3.6 KB |
| __init__.py | -rw-r--r-- | 575 bytes |
| callback.py | -rw-r--r-- | 4.8 KB |
| doubledouble.py | -rw-r--r-- | 10.9 KB |
| general_map.py | -rw-r--r-- | 6.5 KB |
| host_callback.py | -rw-r--r-- | 40.3 KB |
| jet.py | -rw-r--r-- | 19.9 KB |
| loops.py | -rw-r--r-- | 22.6 KB |
| ode.py | -rw-r--r-- | 11.6 KB |
| optimizers.py | -rw-r--r-- | 19.5 KB |
| optix.py | -rw-r--r-- | 16.0 KB |
| stax.py | -rw-r--r-- | 13.2 KB |
| vectorize.py | -rw-r--r-- | 10.1 KB |
