Code demonstrating joint use of jax and numba when calculating model gradients.
This repository has been archived and is no longer maintained. The code is provided for historical reference and may contain unpatched or unknown vulnerabilities. It should not be used in production systems.