Entering and exiting a level should be done via the creation of Interpreter objects

pytorch / functorch

functorch is JAX-like composable function transforms for PyTorch.

https://pytorch.org/functorch/

BSD 3-Clause "New" or "Revised" License

1.38k stars 102 forks source link

Entering and exiting a level should be done via the creation of Interpreter objects #1084

Open zou3519 opened 1 year ago

zou3519 commented 1 year ago

See https://github.com/pytorch/pytorch/pull/90317 for more context. Right now, entering and exiting a level is done via functions but there is additional added complexity that we would benefit from more structure.