Wheelodex » Projects » trainax

trainax

View on PyPI — Reverse Dependencies (1)

0.0.2 trainax-0.0.2-py3-none-any.whl

Wheel Details

Project: trainax
Version: 0.0.2
Filename: trainax-0.0.2-py3-none-any.whl
Download: [link]
Size: 41795
MD5: 6b611b99644ff0771c7401af1435d2e1
SHA256: 402a798beb17534c61ca383ce354a2325c395e8fe6125603a73cdc85f30b8f0c
Uploaded: 2024-10-17 07:29:38 +0000

dist-info

METADATA · WHEEL · RECORD · top_level.txt

METADATA

Metadata-Version: 2.1
Name: trainax
Version: 0.0.2
Summary: Training methodologies for Autoregressive Neural Emulators in JAX built on top of Equinox & Optax.
Author: Felix Koehler
Project-Url: repository, https://github.com/Ceyron/pdequinox
Keywords: jax,sciml,deep-learning,pde,neural operator
Requires-Python: ~=3.10
Requires-Dist: jax (>=0.4.13)
Requires-Dist: jaxtyping (>=0.2.20)
Requires-Dist: typing-extensions (>=4.5.0)
Requires-Dist: equinox (>=0.11.3)
Requires-Dist: optax (>=0.2.0)
Requires-Dist: tqdm (>=4.63.2)
Description-Content-Type: text/markdown
License-File: LICENSE.txt
[Description omitted; length: 9289 characters]

WHEEL

Wheel-Version: 1.0
Generator: setuptools (75.2.0)
Root-Is-Purelib: true
Tag: py3-none-any

RECORD

Path Digest Size
trainax/__init__.py sha256=cCmAuKdygVaHUQhta9xmJPUqAm8Urh_EzD66lmShn0k 371
trainax/_general_trainer.py sha256=zz_VW-vK9_k4kkN8_Ioo3wAx4ea_fsFaS7Z5f2CHjvU 8927
trainax/_mixer.py sha256=sjfyx7hJCYilwORzkqt9TQvVagSbPthfPZiafeOg5Rk 11677
trainax/_sample_data.py sha256=NnNyIPeREwtQb0DEzL8adB1Z85YR_nKtMr8xES1Y5jg 7266
trainax/_utils.py sha256=PUgITtFhZv8cphPKhTeseMCltv5GewlLXGCbQfKTaiM 2027
trainax/callback/__init__.py sha256=-1RnaaUXbVbi4XRCZtfjonAx559bn5K6zadaIoqnQ4s 387
trainax/callback/_base.py sha256=QXCPC1jmE5ulw5S81KHdaPFeTdG48XLu-Pp2Vu2hc-c 1126
trainax/callback/_composite.py sha256=em6diWNNrCh0Y9ig4_pEevNjgtUMS4IY6L4juLgoojc 581
trainax/callback/_get_network.py sha256=tPcC7l7XDmuLh25etaAel5bT3644FbDLb7IZfZShfnM 501
trainax/callback/_grad_norm.py sha256=pAoP8lgmTGEU2ad8pf0qXk3SUvvZPGaRyUu8Cwkko08 2374
trainax/callback/_loss.py sha256=w9Ri2cvqakVJPXmRqd8oyKYRvdMdantXw2m2lZaLWDw 2540
trainax/callback/_save_network.py sha256=Qnv1GDnXTPqgfmXI4A45QhA01AW-r2QAS1pFmqPUEOA 1029
trainax/callback/_weight_norm.py sha256=HkCZhDCADzL7jQXgFF7LnXAk-R-_KY8M-AfVDFi_RIY 1031
trainax/configuration/__init__.py sha256=eQ5qvYfsWAbw43OtuE9-LPx6iRnq0Xp1H4-sCpcOLis 1233
trainax/configuration/_base_configuration.py sha256=y7PD4Ml0zIqYHhNU9TqdzgTHkDKg2H9Lygo5RLbf5W8 1111
trainax/configuration/_composite.py sha256=NQGyoI52Jun0w_hhGkMQq0ONNSt3MOxvSf59K-glE7w 1906
trainax/configuration/_diverted_chain.py sha256=9KLW1iFhoL7PXuhXNZtgdwwWbLmCBalfRl9LcbbNFnk 6715
trainax/configuration/_diverted_chain_branch_one.py sha256=xQeywHuV2mFoQc8AooiIaQx-C_fzJR8GZ2BXuHvwjnk 4888
trainax/configuration/_mix_chain_post_physics.py sha256=Rnvh0PnE0sH2hynzfrW2HBUk-rNYuoTgpcYCcMPh1Wk 5458
trainax/configuration/_residuum.py sha256=npC6e95rT7ASlfNUJvWgt0ttBkgDI2YSulkIZ27sgiI 5132
trainax/configuration/_supervised.py sha256=pH-rVbDXzI8ist44pG9ANmBi8Ffj9nsb4j4gMoH5SFM 4474
trainax/loss/__init__.py sha256=YP3oJf8VPmhtyF1ujfaY5IxqLOWVejJkBXlJDrWQpxI 1080
trainax/loss/_base_loss.py sha256=e0KD3zTugiZqTq1jK40G9Bn8gcNCW-bvxh2BnHO2K3M 3119
trainax/loss/_composite.py sha256=DkeLUlrb7rGx3nZ04aADU9HXXu5mZTf_DBwT0xhzIv4 7
trainax/loss/_lp_loss.py sha256=DkeLUlrb7rGx3nZ04aADU9HXXu5mZTf_DBwT0xhzIv4 7
trainax/loss/_mae_loss.py sha256=a0ke4Sk20GnVS-eYh78GHvDl3q19ksYcBdPXToFKIUg 1397
trainax/loss/_mse_loss.py sha256=k1g6vbzqAIkuOAeUtQcOJbAYiZUlPrvVjMfvP0W7nSA 1398
trainax/loss/_rmse_loss.py sha256=DkeLUlrb7rGx3nZ04aADU9HXXu5mZTf_DBwT0xhzIv4 7
trainax/trainer/__init__.py sha256=eq6E6uNhqeE-Zol7LgNE1Mza1DbIa0FGCwM3erzjlUA 474
trainax/trainer/_diverted_chain_branch_one.py sha256=KbQsXIYImSs13eq9wHwCBJVLnGXuE_TLqKpipAIRTk4 5519
trainax/trainer/_residuum.py sha256=QF0M6PjIIZaMzVlPjrLc4eyOiqdHJ5bKLYaPjiaZvTs 5316
trainax/trainer/_supervised.py sha256=xC6Fx6lckrQLFWx7vS1GDP6SiHpAH8Hndp6oWtGFNHs 5128
trainax-0.0.2.dist-info/LICENSE.txt sha256=lgpCXVtRRnbSt91Qp5nzWoVHEkhJg8MhEIqBJXqzRGQ 1069
trainax-0.0.2.dist-info/METADATA sha256=p2LTstr_Ldt6TgD4wWtHRzIhRSIujL5m_FYpYvsXcEE 9883
trainax-0.0.2.dist-info/WHEEL sha256=OVMc5UfuAQiSplgO0_WdW7vXVGAt9Hdd6qtN4HotdyA 91
trainax-0.0.2.dist-info/top_level.txt sha256=_vGo--Df2J6eN6MXDZ9GvGT7Walmkx63GEY3WR1BA30 8
trainax-0.0.2.dist-info/RECORD — —

top_level.txt

trainax