Expand CUDA tape support
- Add support for
kwargs
static_args : This is a bit funny since we are interpreting the function and not compiling.
abstracted_axes
- Test and add support for other devices.
- control flow operations.
- quantum control of arbitrary gates.
- adjoint operations should create their own parameters.
- Determine version compatibility.
- Fix samples
- Fix state
- Better error messages for these four use cases:
- expval(float * obs) (get vague JAX error)
- expval(obs + obs) (get vague CUDA quantum error)
- gradients (get vague JAX error)
- counts/sample without shots specified (get a vague JAX error)
- AOT compiling
- Caching