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=Ks7wG1hvY5RgXcExMJoMAafl8txg91u5AnBPiKYZqrs,14543
folx/interpreter.py,sha256=ILRxT6dS4P06sH6a8L9wbh36O2Lhl0kLqtPVteLyPcU,12702
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=RRVQhvIX3RxU9qHfKGZVl9ZatAEC64sRE7P3QGDq9hk,4785
folx/experimental/pallas/attention/custom_gradients.py,sha256=rqyhe2jxiOlxpqDPKmP7P-C8VNamss6be17TI8PEvrM,22871
folx/experimental/pallas/attention/forward_laplacian.py,sha256=2Jn6EHU7jQCiZA6IszK0oMB1S9nV2a7BcIWwppNFs1g,32152
folx/experimental/pallas/attention/mhsa.py,sha256=ysYlNoSGoAkC05rmwoX-l9n8st02IQJrAAUDy9NGYMc,4844
folx/experimental/pallas/attention/mhsea.py,sha256=G6Aa6hG5EDrcPLM0GSQOFHv6jvxbeg-eDBTRjCNVdcw,5117
folx/experimental/pallas/attention/utils.py,sha256=BgGAhHjZ0aED0sD0FiuouW5cZ4FM8oogIP7ri4EIyNI,8224
folx-0.2.18.dist-info/METADATA,sha256=KxIptuhv4WwFFi2oqtjk8-vaBVZysePbbVzs4iT4wzc,12962
folx-0.2.18.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
folx-0.2.18.dist-info/licenses/LICENSE,sha256=ws_MuBL-SCEBqPBFl9_FqZkaaydIJmxHrJG2parhU4M,1141
folx-0.2.18.dist-info/RECORD,,
