folx/__init__.py,sha256=fQY_KCiTGiOqyDEpchPXoplPQdREzmbxeGzsxLtAMXM,743
folx/ad.py,sha256=orMa8f7iM_alP2lcsWbrS6gJfOYktE5r3vlDltQvUsU,2812
folx/api.py,sha256=l9KqdhAloBe8ZePHxBfpGHrr8xYN4LajbDkEifrXbik,14057
folx/custom_hessian.py,sha256=vzy11r9lKShKPZnxsdZaUzENIzTVEJLG6S42ehIiwHs,3804
folx/hessian.py,sha256=CGpaAPqtMZyxjpj-pB_yPXYkmk00gqOQuUJSrf31fA4,14595
folx/interpreter.py,sha256=spVOxVIAuP637swUvLKNCPhroCSqyUC_Q6C3LmWrI80,12693
folx/jvp.py,sha256=mg20K89eALa02YcVflT3O0cWiF6zdpj9QalAe9bqM2A,21240
folx/operators.py,sha256=azXWxJx3s0sel0rKZpznS9lCcKS6Pvf1KSmBPkGLrWM,1972
folx/tree_utils.py,sha256=LTzEvNrJvAE6KGrhUFJnaSD0Nmt4wicogIO5idlSCxI,2225
folx/utils.py,sha256=j-cbQs64zXxmv81fFL5GIdAIUEqWqD5F5eozPh6FOfg,18726
folx/vmap.py,sha256=0P-biiPSDCTI8Vwlgvk8V7MbYyNu1GFnt4Pg0PIlu70,2844
folx/wrapped_functions.py,sha256=-N5Od6vq2fxIUxzpJjQM-X7NlDknI3uQjEbw-igU-FQ,19631
folx/wrapper.py,sha256=Lg1kyTzNjjbhtHW-N7Sloff-iaPcoUU_oQLG2EJE0xI,5241
folx/experimental/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
folx/experimental/vmap.py,sha256=8eDek-sSfw7CqEuCWlj_vH7pN9RLrte3gp_FRub6iX4,2362
folx/experimental/pallas/attention/__init__.py,sha256=3Q5kM1O0ggxeUSxGZDPSl7NMsBPT49A_sGcQ9hnL6Gw,2791
folx/experimental/pallas/attention/custom_gradients.py,sha256=iHGdDIWas7VZRzVM_6wgDIwc7xWSnovBtwRgJ6eMdJo,7758
folx/experimental/pallas/attention/forward_laplacian.py,sha256=Z3D5mTw1s-kEGbGPI2FCZK4viueEUPPhKlCDt_bWlYs,17826
folx/experimental/pallas/attention/mhsa.py,sha256=2w-u7GVNHhUVhcgUAk-GCIqKAAsa7BRQkGKXVQ2I8SQ,4855
folx/experimental/pallas/attention/utils.py,sha256=Bj2vcydleYtDhFq1clKB9GvEiyA-HU041a1frHciDlM,6670
folx-0.2.16.dist-info/METADATA,sha256=9PQ0JVPCBCU4w-TjFN1zJB_lW90QBoew91eu3ruPQ-4,12962
folx-0.2.16.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
folx-0.2.16.dist-info/licenses/LICENSE,sha256=ws_MuBL-SCEBqPBFl9_FqZkaaydIJmxHrJG2parhU4M,1141
folx-0.2.16.dist-info/RECORD,,
