Dougal Maclaurin c36e1f7c1a Make trace dispatch purely a function of context rather than a function of both context and data. This lets us delete a lot of machinery for managing data-dependent tracing: levels, sublevels, post_process_call, new_base_main, custom_bind and so on. ...

PiperOrigin-RevId: 691086496

2024-10-29 11:04:31 -07:00

..

__init__.py

Make trace dispatch purely a function of context rather than a function of both context and data. This lets us delete a lot of machinery for managing data-dependent tracing: levels, sublevels, post_process_call, new_base_main, custom_bind and so on.

2024-10-29 11:04:31 -07:00

linalg.py

Activate the FFI implementation of SVD on GPU.

2024-10-17 17:57:06 -07:00