Wheelodex » Projects » mistral-v0.2-jax

mistral-v0.2-jax

View on PyPI — Reverse Dependencies (0)

0.0.1 mistral_v0.2_jax-0.0.1-py3-none-any.whl

Wheel Details

Project: mistral-v0.2-jax
Version: 0.0.1
Filename: mistral_v0.2_jax-0.0.1-py3-none-any.whl
Download: [link]
Size: 23472
MD5: 622b697f37fe4f609463c2634c5cd503
SHA256: 95b6c0443deb83ed3d7e15a9ddf78c893c9f55038c287c662ab4afa407651b7c
Uploaded: 2024-04-17 13:36:15 +0000

dist-info

METADATA · WHEEL · RECORD · top_level.txt

METADATA

Metadata-Version: 2.1
Name: mistral-v0.2-jax
Version: 0.0.1
Summary: JAX implementation of the Mistral v0.2 base model.
Author-Email: Shin <shin[at]yixiaoer.sg>
Project-Url: Repository, https://github.com/yixiaoer/mistral-v0.2-jax
Project-Url: Issues, https://github.com/yixiaoer/mistral-v0.2-jax/issues
Classifier: Development Status :: 4 - Beta
Classifier: Programming Language :: Python
Classifier: License :: OSI Approved :: MIT License
Requires-Python: >=3.12
Requires-Dist: jax
Requires-Dist: jaxlib
Requires-Dist: torch
Requires-Dist: transformers
Requires-Dist: datasets
Requires-Dist: einops
Requires-Dist: einshard (<0.3,>=0.2.0)
Requires-Dist: sentencepiece
Requires-Dist: accelerate
Requires-Dist: mypy
Requires-Dist: sphinx
Requires-Dist: sphinx-autobuild
Requires-Dist: sphinx-autodoc-typehints
Requires-Dist: sphinx-book-theme
Description-Content-Type: text/markdown
License-File: LICENSE
[Description omitted; length: 6185 characters]

WHEEL

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

RECORD

Path Digest Size
mistral_v0_2/__init__.py sha256=DsAHdxLC16H2VjdFOU5tBx2xT9VnNQ-XbTS24fRCa_w 22
mistral_v0_2/lib/__init__.py sha256=uG9Qx64PTIPRBu2-tBU5_vEijJWlztk5pwVKSGQL_Nw 107
mistral_v0_2/lib/array_conversion.py sha256=ddmD80_nOJ5wjwp28HN9XYIjeoytD9pJBhQbO71-nGc 2416
mistral_v0_2/lib/collate_fn.py sha256=npt2dOFIa72pAxobBZZlUw8jBUgHRqURgBlXsbSwCE4 4627
mistral_v0_2/lib/example_data.py sha256=U37F4OmQ0_JWD56R2EBLkvYFolPJq7k7zhQrTMuKMvc 1411
mistral_v0_2/lib/generate.py sha256=Df0Bzw2x5oA5AvkGZuhARW-w4jOkxh_-CLaZhnDM7ME 14764
mistral_v0_2/model/__init__.py sha256=r7djSv2DfhsJsU7P6xeogOgysT6VmXxuQkJcPXdL77w 253
mistral_v0_2/model/attention.py sha256=znSQCDJiepUknffPsgRktknu-WPQH8bKAbbDO9OBQLo 6600
mistral_v0_2/model/decoder.py sha256=tHi8jiJVP5vuDJqso5PRaFd1XvjBz_zyqgDiZXOLilc 2292
mistral_v0_2/model/decoder_block.py sha256=paRL_ml-4tERkPn73I2187Jj2L3uKSI5tp5VXAvamFQ 3702
mistral_v0_2/model/embedding.py sha256=KD764ngBbDlVS2oCWhRyUN2cP4p-yzUCbqXz8A6qCbQ 2218
mistral_v0_2/model/kvcache.py sha256=QEQIQjURUkgdvstggs7hsxFgEutkd8GImSYIURkzjig 70
mistral_v0_2/model/mistral_lm.py sha256=tV2a79MmkWObEK7TUIRGeoU6jkBWATKiqWExE8lDDak 4128
mistral_v0_2/model/mistral_model.py sha256=6ZUPplYJz9l5P1yBcdx1nFsVuBPCxCau66xx07Decgo 2807
mistral_v0_2/model/mlp_layer.py sha256=dK1v650ME7ymeL54YzAKCmznoX5vjKKMJ8ul7QuZTBw 2141
mistral_v0_2/model/rms_norm.py sha256=cLY7tYyOOkR_i1ptrsCCS1XMBol4Av7NiflZMgYHKEg 2465
mistral_v0_2/model/rotary_embedding.py sha256=1fVSRw7E86AsQb0LKiFwz5pXuPFkYreqYNs2Ly1Ij4c 2996
mistral_v0.2_jax-0.0.1.dist-info/LICENSE sha256=AWbEMAUVzy0fTxQvntgL0ST6cIBLVeM0IxqbOMQ7iOw 1065
mistral_v0.2_jax-0.0.1.dist-info/METADATA sha256=9FLERyjh3Vulktj1SajW8DMQ2niQi7dqX7uSW-tNHKk 7085
mistral_v0.2_jax-0.0.1.dist-info/WHEEL sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ 92
mistral_v0.2_jax-0.0.1.dist-info/top_level.txt sha256=yzPGl28b0bzu6YM3F2ZN_gVAAZ4vH-fasnPJ65rVdJY 13
mistral_v0.2_jax-0.0.1.dist-info/RECORD — —

top_level.txt

mistral_v0_2