tjax/__init__.py,sha256=4TJcxPEyrVrwJyYRbSIZXgcrJ7hY8OWeBmNcpQB9oe8,4506
tjax/_src/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
tjax/_src/abstract_method_decorators.py,sha256=Dtlp_lFqvgAYvRkVUJD2bQbQBS8V7u_Lx5hnReCUZsk,6270
tjax/_src/annotations.py,sha256=Z1vZ4TYW0A6sIXsiadDocczAKruldYsgG7T4NfpPJxc,1529
tjax/_src/cotangent_tools.py,sha256=cwLznUSr9F-aO5n6ajxCFqQcECKb_jHQ74_7-gbAebA,3517
tjax/_src/dataclasses/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
tjax/_src/dataclasses/dataclass.py,sha256=vw2rMdI0bGZdRr7zjjoaw2QlenP_J4ROK5f0wPoyb4Y,5801
tjax/_src/dataclasses/helpers.py,sha256=QRHi2YHSPRQsBEF9vN_wzRsC0-BNpzfXXBddCZksz2I,6689
tjax/_src/display/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
tjax/_src/display/colors.py,sha256=n2iou6ApMVLZAhJ1MIlutE82g1Nq3lQN7wJHdnQJ2i4,441
tjax/_src/display/display_generic.py,sha256=HxOhUHuH5075cGfg03ox_cuaZ_q68hd5hFpCci_cxKI,10297
tjax/_src/display/generic_string.py,sha256=h0igFZCrECGTPcFhiV_WnTmF1tqG210wmmXwD_-NoX0,985
tjax/_src/display/internal.py,sha256=G53op9tiJn9pgQtmr9HAFIC8PHtKO-3y9bnoGFM1Q04,1322
tjax/_src/display/print_generic.py,sha256=-mtYtnbDkDtgtVHYWc6ZLihJfFFC2lbmCijsrO6aGZE,2070
tjax/_src/dtype_tools.py,sha256=LG-6ADZelGO8AzV68OcaDiihaIMwT3VtYt4FgFIXLrE,818
tjax/_src/frozendict.py,sha256=K-rc1j9bid9nYHCqMPMlX2m2QSsl4WUilhE1xTbPJ54,2988
tjax/_src/function_markers.py,sha256=cFlrc-wn61o3CZeV7wBBXhjteliflrhlDZbn7cmB4FE,371
tjax/_src/gradient/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
tjax/_src/gradient/accumulation.py,sha256=wG1NjoqFTvjqQgowXKYJ-Vn9_h2NISj4aP5ebA-i3b0,4487
tjax/_src/gradient/adding.py,sha256=7B1DlfLexbB_-3hPBxKXt0Sh4DQN02PzPwmSd1_UfWU,1972
tjax/_src/gradient/aliases.py,sha256=imqttkh99Gg8Yf-Qm5I8LYLmX_8me6LdNhqxLDhZ834,40859
tjax/_src/gradient/basic.py,sha256=XJAJa2yx1WlSZYr1nGHB_3Ei33KB_eiFS-FO1O2qYNs,1844
tjax/_src/gradient/clipping.py,sha256=qSiDl4Oexcft7mqQEdgjVl_sIC6fsKpSueoIjnV6F4I,2710
tjax/_src/gradient/combining.py,sha256=2GIGWdPz7B0WpcSFFvOgRJXs-Oww19kWue9YWfMnrow,8433
tjax/_src/gradient/conditionality.py,sha256=nxVUyYU8C0JkDBkvtNOsWBkT-Q2iKD2eYYwGDq-R3nM,2831
tjax/_src/gradient/constraining.py,sha256=NCMBHwHFv8PC7CpU3T4x-FwPXlaWkafbZws_60ewaPo,916
tjax/_src/gradient/contrib.py,sha256=m72yeXdZme6lppebVsaxNB-YPp9x2LXk9lxnrV2H55g,56536
tjax/_src/gradient/linesearch.py,sha256=kEjt6cjuXLGUfc0CE5xosF1nh96txE0ZzZKkMHHMOuk,5871
tjax/_src/gradient/monitoring.py,sha256=icM9GLA9D_s9F4bAneoe9E2JemSGJQF_jQWc4wwQ9rw,2807
tjax/_src/gradient/scaling.py,sha256=LN1Lrdo8kp3VlZvSj1CHyy2lbRW8MXEQ2sKPbOYE9aQ,25656
tjax/_src/gradient/schedule.py,sha256=NwkKaStZva7YVzkGB7_QR2x7Bnsftt-oGq7L77OfrTA,902
tjax/_src/gradient/smd.py,sha256=dDb6Uy7OF0nGKZSt8KIx6n_H_C9ZDw9VNRWIVH3WSrM,2508
tjax/_src/gradient/transform.py,sha256=or5S9D-RkueBQItz8hQ21sg7_KrNVjnBdiKlP01Jwlc,11001
tjax/_src/graph/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
tjax/_src/graph/display.py,sha256=m6eK4br-Kg_Z-WYV9gZ-TK7KZtFH7D8Q1eJ9Z6q_7nE,1319
tjax/_src/graph/register_jax.py,sha256=q8BYWCsPvZL3DR_yGCpN98aWB_BmnKAGI-YscJ7D-3s,3082
tjax/_src/graph/types_.py,sha256=zymhYpKnREWFClCA20EBh3pk4qtp-NIikTAt94ONVvs,2083
tjax/_src/leaky_integral.py,sha256=sM_XcDIpP6yWaJcOfhlMLJN-KoFVFhmWlu8sy6GY13Y,4127
tjax/_src/math_tools.py,sha256=sjdTEwmniYgDA901gY_g6swcJTMTe8T-NTUP1ohgnew,11905
tjax/_src/project_tree.py,sha256=wZuZ0GHq89lFdmqmj4Vm7lz7L7ksnUBJorpqWNbaWds,3312
tjax/_src/rng.py,sha256=BYsl3D_-BYAQQ-wsXJ2rXr7y3GSZk8nerLU-p7slqj4,1866
tjax/_src/shims.py,sha256=q6cVgqlqTU0X0xtS5B1kLzYeFSBF1i7J3J4l7To6QVQ,6699
tjax/_src/testing.py,sha256=fFh_XdWQYbdmGx0aSPAN1l_EzsI9Fgzsrd-_LWJoYZk,9969
tjax/_src/timer.py,sha256=EETL6XnrUH_BsbWEYap1mXjy3Di4idit7YLDk9AYwEU,2377
tjax/_src/tree_tools.py,sha256=OtGqBHmkntdP2e_viqvgZbgXtwy3_nDyKIlc0sM7yD0,2169
tjax/dataclasses.py,sha256=PD2BfnMdZh8eg7_nTc6msJoNE5DT3IQ2gV55O26PHiE,246
tjax/gradient.py,sha256=GVjcdNxFNP2oD2VxbSXHosVe7DTfhHrRNKzdDcBqd9U,5742
tjax/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
tjax-2.0.0.dist-info/licenses/LICENSE,sha256=xVlGM_ARkYfXpSNBj65KNp1-yutURh3zd08e8ilPBic,11343
tjax-2.0.0.dist-info/WHEEL,sha256=3NO1oYsF7mANFUWVdyEQLM-zmecczWWEmmF5rEbXrWk,80
tjax-2.0.0.dist-info/METADATA,sha256=Ik8GaJiBe0pCRoBDD6aYIlFDxlO_JAr2VHZ29rZC1iI,5287
tjax-2.0.0.dist-info/RECORD,,
