torch_cudagraph_fx is a make_fx-style tracer and interpreter oriented to graph capture for CUDA-graphable training steps. It is a prototype of the design in DESIGN.md. Your code must be CUDA-graphable ...
This project builds a multi-class disease classifier using symptom data. The trained model can predict the most likely disease given a set of symptoms and return a recommended treatment, dynamically ...