Jesse Michel

Hi @croci, I believe so. I just tested it by adding: ``` x = jax.grad(add_one)(.1, .1) x = jax.value_and_grad(add_one)(1., 2.) ``` to the end of the test.py file. --Jesse

Hi @pi314ever, Thank you for the good question! We currently only support CLI style files using the `config=[...]` syntax https://github.com/swansonk14/typed-argument-parser?tab=readme-ov-file#loading-from-configuration-files. You can load from a JSON file via https://github.com/swansonk14/typed-argument-parser?tab=readme-ov-file#load. We'd...

Hi @alecjacobson and @squidrice21, Thank you for the wonderful question. 1. Why is the answer wrong and what's a way to check this? The derivative violates the transversality condition and...

@gilbo I'm talking explicitly about the case of `f([x < t])`, where the degeneracy arises from the chain rule. I'm not talking about taking a pair of arbitrary conditionals and...

@alecjacobson, To answer your question directly: > What's a good rule I can follow to be sure that I provide valid input to Teg? No nested IfElse? For the current...

@marzia-riso, Thanks for taking a look at our repo. Does the fix work for you? Let us know if you have other issues with the repo. --Jesse

Hi @yyuting, Thanks for taking a look at the repo, I believe the C code that we generate is CUDA C. You can call it with: ```python evaluate(expr, backend='C') ```...

Remove `TegVar` and implement the appropriate static analysis to accommodate for the change

Unfortunately, these warts are part of the price that we pay for living in Python rather than writing a parser. You can use `b = (1 < x) & (x...

As a result, I'd view this less as a bug and more as a factor in deciding what our frontend should be in the end.