nextgenjax

View on PyPIReverse Dependencies (0)

0.2.0 nextgenjax-0.2.0-py3-none-any.whl

Wheel Details

Project: nextgenjax
Version: 0.2.0
Filename: nextgenjax-0.2.0-py3-none-any.whl
Download: [link]
Size: 24067
MD5: 3b64e99ba07f4595dfa1129d1dc95b0c
SHA256: 43f6e24ae213417a2bd09bf8fef7598d90851ed48d8b3a6feedbeee95f84ec89
Uploaded: 2024-07-23 15:35:15 +0000

dist-info

METADATA

Metadata-Version: 2.1
Name: nextgenjax
Version: 0.2.0
Summary: A JAX-based neural network library surpassing Google DeepMind's Haiku and Optax
Author: kasinadhsarma
Author-Email: kasinadhsarma[at]gmail.com
Home-Page: https://github.com/VishwamAI/NextGenJAX
License: MIT
Classifier: Programming Language :: Python :: 3
Classifier: Programming Language :: Python :: 3.9
Classifier: Programming Language :: Python :: 3.10
Classifier: Programming Language :: Python :: 3.11
Classifier: License :: OSI Approved :: MIT License
Classifier: Operating System :: OS Independent
Requires-Python: >=3.9
Requires-Dist: Pillow
Requires-Dist: chex (>=0.1.86)
Requires-Dist: dm-haiku
Requires-Dist: fairscale
Requires-Dist: flax
Requires-Dist: gym
Requires-Dist: jax
Requires-Dist: jaxlib
Requires-Dist: moviepy
Requires-Dist: numpy (>=1.24.1)
Requires-Dist: optax
Requires-Dist: scipy
Requires-Dist: stable-baselines3
Requires-Dist: sympy
Requires-Dist: tensorflow
Requires-Dist: torch
Requires-Dist: transformers (==4.38.0)
Requires-Dist: whisper
Requires-Dist: flake8 (==3.9.2); extra == "dev"
Requires-Dist: pytest (==6.2.4); extra == "dev"
Provides-Extra: dev
Description-Content-Type: text/markdown
License-File: LICENSE
[Description omitted; length: 6248 characters]

WHEEL

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

RECORD

Path Digest Size
nextgenjax/__init__.py sha256=53YIeTEM2RzAwXPuLZhyEFhxyl9-GjCivxIFNRUTook 502
nextgenjax/activations.py sha256=NP5fYdmr9CkxyT-0tjHsQNhJvKVVaLCFw_53zJqe-JM 1587
nextgenjax/custom_layers.py sha256=chjvlqJ2vYrGtkAH-IVOxXZkF0lzAhg37pt2adULzuM 2260
nextgenjax/generation_capabilities.py sha256=lTAAygE5lTVvsOL4-J1OEzB9TqAFSDeyFz91bYJ1Crg 950
nextgenjax/hybrid_model.py sha256=-FunX_iAKiHdy3bRdHGH5IiBU3vDdVHpIk2pkGimrZU 1078
nextgenjax/integrations.py sha256=i9v9cdeLpFpYovktrORqoTpHkbfkwNjXrHeSWfTf8Co 1720
nextgenjax/jax_integration.py sha256=5I-8D--8V_8Zk_pqTGDd-lCA77GLlUs8TdWUrk1x-RU 2246
nextgenjax/layers.py sha256=MViadJtEvj1K2TNs9i5whCVf8Xwid7TfiLRx3G-Vk2M 3813
nextgenjax/model.py sha256=XgqrTm1dQdbWj31QbE9JlolFzkGiJgh1w7neAhxnLyo 8544
nextgenjax/optimizers.py sha256=f_X5VP1_cLu2Ex8XReXcDtn3LKncmsY3XZqv8RsZuOU 5142
nextgenjax/parallelism_example.py sha256=ECzTGqCapsLKH5SQPrNRTf0kiBCv0MLGbQdAJf47FK4 4717
nextgenjax/train.py sha256=WBhsqw6zRHFjyMX7YkWoKaga9-uTtRiP_q9vIL_hy60 19103
nextgenjax/transformer_models.py sha256=doTpjvuwE9oEDJ-NLUc8dwwXEjg_efKds4d-G62yr_w 3760
nextgenjax/verify_compatibility.py sha256=GVi186JWW0W2VKbdLn2NZsrDS-R8IcDh6n5n9kElK8I 957
nextgenjax/vocab_parallel_embedding.py sha256=qHdrLut-TbXjVUHKU6BETOip0LINc3qMTQaWUqDfR5I 2096
nextgenjax-0.2.0.dist-info/LICENSE sha256=UIp30ue1HZit7tMmSK0SS3swJBqOcLLnLJn5LY5YdNE 1036
nextgenjax-0.2.0.dist-info/METADATA sha256=oDStlP7FkCo2PlIshN9QIHiCuJ5clupvOhZCqPIVMNA 7477
nextgenjax-0.2.0.dist-info/WHEEL sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA 92
nextgenjax-0.2.0.dist-info/top_level.txt sha256=gpFQXU9X1aI80kySF6nBaCHj1_n4CbK_l8i83ptGSSE 11
nextgenjax-0.2.0.dist-info/RECORD

top_level.txt

nextgenjax