Wheelodex » Projects » tax-dp

tax-dp

View on PyPI — Reverse Dependencies (0)

0.1.1 tax_dp-0.1.1-py3-none-any.whl

Wheel Details

Project: tax-dp
Version: 0.1.1
Filename: tax_dp-0.1.1-py3-none-any.whl
Download: [link]
Size: 51396
MD5: fed388db2985412abfa722636dbd60bd
SHA256: 6b2cd0eeb198ab00ab3e53af03e2e0191868526b9b7c38c96960e082856f45a0
Uploaded: 2024-10-07 15:27:59 +0000

dist-info

METADATA · WHEEL · RECORD · entry_points.txt

METADATA

Metadata-Version: 2.1
Name: tax-dp
Version: 0.1.1
Summary: A fully sharded data parallel trainer for jax
Author-Email: Rares Dolga <rares.dolga.16[at]ucl.ac.uk>
License: MIT
Requires-Python: >=3.10
Requires-Dist: jax[cuda12] (==0.4.31)
Requires-Dist: tqdm (==4.66.2)
Requires-Dist: wandb (==0.17.2)
Requires-Dist: black (>=24.8.0)
Requires-Dist: flax (==0.8.4)
Requires-Dist: orbax-checkpoint (==0.6.4)
Requires-Dist: matplotlib (>=3.9.1)
Requires-Dist: pip (>=24.2)
Requires-Dist: jaxtyping (>=0.2.34)
Requires-Dist: torch (>=2.4.1)
Requires-Dist: transformers (>=4.45.1)
Requires-Dist: datasets (>=2.19.2)
Description-Content-Type: text/markdown
[Description omitted; length: 585 characters]

WHEEL

Wheel-Version: 1.0
Generator: pdm-backend (2.4.1)
Root-Is-Purelib: true
Tag: py3-none-any

RECORD

Path Digest Size
tax/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
tax/config.py sha256=_1lNmDVoDL52PIr_EJHEIzkTwGjGca5fs1ZUJcKhlqI 5189
tax/evals/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
tax/evals/base.py sha256=CnpOXw69ioDvcHjry6mD1Uek0MPpuq3ebqZJk2HRLDQ 2200
tax/evals/class_eval.py sha256=AUPDKMt_RYm4eyIteiaeATvMp4MR-uoY2LI2NabyPso 1332
tax/evals/lang_eval.py sha256=xhgvwY0gX_rtIWF08Io3tDMAVaGaP54Ly2Spug2UWuU 3202
tax/evals/losses.py sha256=DNC6t54HcO6lxuw9LSKPxEPMw4nz7fLs02FD_-C80No 1177
tax/examples/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
tax/examples/bin/download_lra_data.sh sha256=EPZcIrQBHhZ5M2prSjhPSavqtOdbDhhQwLreDNN35Nc 723
tax/examples/bin/run_lm.sh sha256=lt4F3Bs7kniXRO4VcRJ4RHjebuycYR5-hRXCfUVJCMo 634
tax/examples/bin/run_lra.sh sha256=Dp-edg0Q-2qP4UHsMfbtLQ4lxsCLhCcAedebomqSVk0 591
tax/examples/configs/imdb.yaml sha256=CGwmPpdVWSyDh05KIs8IdZNcdeUMbUqfe4WD-Ynl8wY 542
tax/examples/configs/lm.yaml sha256=DYN_8B-c16uxlYaaZNqLY887PJ4TVvxnUWEGh03ijns 999
tax/examples/gemma/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
tax/examples/gemma/exp.py sha256=XVyvbR94fYfYrZXpM_b07cIs-v35HT1ctPJR7lrsbIc 5975
tax/examples/gemma/gemma.py sha256=rctylh6ALy-rWfI6gpd4ZtKjJLyb-NK9pTND3gRLzUA 17954
tax/examples/gemma/lm_dp.py sha256=_fjm7mch9QSsxt5l_Pn-q2pBmMsgVhERWXpek6DEhjs 7602
tax/examples/gemma/task.py sha256=dDq7RiLMaZ7QLtvdNV0yuTHREFuf6cmIf3uaf77Xu9A 3851
tax/examples/lra/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
tax/examples/lra/exp.py sha256=sbaMSMFMG6fY2LHQUPRBvuvvdAdWzemUGBzheCwBVjk 5486
tax/examples/lra/lra_dp.py sha256=yjp3SKjrCzoNE3BxiygcW8Yg3jxZl8D32yXsyPzg7Oc 11667
tax/examples/lra/lra_tok.py sha256=LI-sES_78OLWyf2jmPiXFSWEquS6z7aWrg7zRLmeLEE 17520
tax/examples/lra/model/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
tax/examples/lra/model/attention.py sha256=7goqgoQ7hwx0nrmBCNFxr2y6KJj1ALNZ6Uhv5RQJXpU 3246
tax/examples/lra/model/lra_enc.py sha256=8vZQpIDunqcNPDfscvrx-RTBgzQCRR0gQz2crBzw1Rw 6539
tax/examples/lra/model/task.py sha256=R-8t8tekkZal_XNxr9jSjdry4zHpr0cBmFVAvXHsffw 5005
tax/examples/utils.py sha256=avf5w7UuRZmj0WTA_IQTj0Xe8kRbCgdl9GZJK3mGV-Q 880
tax/lr_schedules.py sha256=F6CPnacukOlGtbXX0io43S53VgW4MjnzhzlwO78RrOQ 512
tax/train/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
tax/train/dfsp_jax.py sha256=QOY_bzcrlt4CV9Ra5kOrt2aisLkw_0iULCOQF6NhL-o 30749
tax/train/jax_single_host.py sha256=f6uvgH-CzqVKO1Wza35cNRunRXtQ9YKP5Tk4KPpsK6M 24326
tax/train/utils.py sha256=C7bomY3Gq5AYdGG1wLsPRw_ApZEDe8_29jWsnzsYAZA 666
tax_dp-0.1.1.dist-info/METADATA sha256=qs_CtPPpeeEXgYQ_GIzip-cqos0Gz-b89KWNSbHhyk8 1195
tax_dp-0.1.1.dist-info/WHEEL sha256=Vza3XR51HW1KmFP0iIMUVYIvz0uQuKJpIXKYOBGQyFQ 90
tax_dp-0.1.1.dist-info/entry_points.txt sha256=6OYgBcLyFCUgeqLgnvMyOJxPCWzgy7se4rLPKtNonMs 34
tax_dp-0.1.1.dist-info/licenses/LICENSE sha256=XoBm14Y5-XGRqmoEY8m1a-9Y8XRlaj8jkKFn9wZ5_dY 1068
tax_dp-0.1.1.dist-info/RECORD — —

entry_points.txt

[empty]