pasteurlabs/tesseract-jax
Execute + differentiate Tesseracts as part of JAX programs, with full support for function transformations like JIT, grad, and more. ⚡
GitHub repository with 32 stars and 2 forks.
Language: Python
Topics: automatic-differentiation, differentiable-programming, jax, remote-procedure-call, scientific-machine-learning, tesseract