autograd_gamma_regularized_incomplete_gamma_function_derivative.py

1import autograd.numpy as np
2from autograd import grad
3from autograd_gamma import gammainc
4
5# Define a function using the regularized lower incomplete gamma function
6def f(a, x):
7    return gammainc(a, x)
8
9# Compute the gradient of the function with respect to the first argument 'a'
10grad_f = grad(f, 0)
11
12# Evaluate the gradient at specific values
13print(grad_f(1.0, 2.0))