lalamo 0.5.13__tar.gz → 0.5.14__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (110) hide show
  1. {lalamo-0.5.13 → lalamo-0.5.14}/PKG-INFO +5 -5
  2. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo/__init__.py +1 -1
  3. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo.egg-info/PKG-INFO +5 -5
  4. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo.egg-info/requires.txt +4 -4
  5. {lalamo-0.5.13 → lalamo-0.5.14}/pyproject.toml +4 -4
  6. {lalamo-0.5.13 → lalamo-0.5.14}/LICENSE +0 -0
  7. {lalamo-0.5.13 → lalamo-0.5.14}/README.md +0 -0
  8. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo/common.py +0 -0
  9. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo/data/__init__.py +0 -0
  10. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo/data/huggingface_message.py +0 -0
  11. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo/data/lalamo_completions.py +0 -0
  12. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo/data/utils.py +0 -0
  13. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo/main.py +0 -0
  14. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo/message_processor.py +0 -0
  15. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo/model_import/__init__.py +0 -0
  16. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo/model_import/common.py +0 -0
  17. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo/model_import/decoder_configs/__init__.py +0 -0
  18. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo/model_import/decoder_configs/common.py +0 -0
  19. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo/model_import/decoder_configs/executorch.py +0 -0
  20. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo/model_import/decoder_configs/huggingface/__init__.py +0 -0
  21. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo/model_import/decoder_configs/huggingface/common.py +0 -0
  22. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo/model_import/decoder_configs/huggingface/gemma2.py +0 -0
  23. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo/model_import/decoder_configs/huggingface/gemma3.py +0 -0
  24. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo/model_import/decoder_configs/huggingface/gpt_oss.py +0 -0
  25. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo/model_import/decoder_configs/huggingface/lfm2.py +0 -0
  26. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo/model_import/decoder_configs/huggingface/llama.py +0 -0
  27. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo/model_import/decoder_configs/huggingface/llamba.py +0 -0
  28. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo/model_import/decoder_configs/huggingface/mistral.py +0 -0
  29. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo/model_import/decoder_configs/huggingface/modern_bert.py +0 -0
  30. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo/model_import/decoder_configs/huggingface/qwen2.py +0 -0
  31. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo/model_import/decoder_configs/huggingface/qwen3.py +0 -0
  32. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo/model_import/huggingface_generation_config.py +0 -0
  33. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo/model_import/huggingface_tokenizer_config.py +0 -0
  34. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo/model_import/loaders/__init__.py +0 -0
  35. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo/model_import/loaders/common.py +0 -0
  36. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo/model_import/loaders/executorch.py +0 -0
  37. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo/model_import/loaders/huggingface.py +0 -0
  38. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo/model_import/loaders/utils.py +0 -0
  39. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo/model_import/model_specs/__init__.py +0 -0
  40. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo/model_import/model_specs/common.py +0 -0
  41. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo/model_import/model_specs/deepseek.py +0 -0
  42. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo/model_import/model_specs/essential_ai.py +0 -0
  43. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo/model_import/model_specs/gemma.py +0 -0
  44. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo/model_import/model_specs/gpt_oss.py +0 -0
  45. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo/model_import/model_specs/huggingface.py +0 -0
  46. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo/model_import/model_specs/lfm2.py +0 -0
  47. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo/model_import/model_specs/llama.py +0 -0
  48. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo/model_import/model_specs/llamba.py +0 -0
  49. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo/model_import/model_specs/mirai.py +0 -0
  50. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo/model_import/model_specs/mistral.py +0 -0
  51. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo/model_import/model_specs/pleias.py +0 -0
  52. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo/model_import/model_specs/polaris.py +0 -0
  53. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo/model_import/model_specs/qwen.py +0 -0
  54. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo/model_import/model_specs/reka.py +0 -0
  55. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo/models/__init__.py +0 -0
  56. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo/models/classifier.py +0 -0
  57. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo/models/common.py +0 -0
  58. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo/models/language_model.py +0 -0
  59. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo/modules/__init__.py +0 -0
  60. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo/modules/activations.py +0 -0
  61. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo/modules/classifier.py +0 -0
  62. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo/modules/common.py +0 -0
  63. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo/modules/decoder.py +0 -0
  64. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo/modules/embedding.py +0 -0
  65. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo/modules/linear.py +0 -0
  66. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo/modules/mlp.py +0 -0
  67. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo/modules/mlx_interop.py +0 -0
  68. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo/modules/normalization.py +0 -0
  69. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo/modules/rope.py +0 -0
  70. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo/modules/token_mixers/__init__.py +0 -0
  71. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo/modules/token_mixers/attention.py +0 -0
  72. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo/modules/token_mixers/common.py +0 -0
  73. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo/modules/token_mixers/mamba.py +0 -0
  74. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo/modules/token_mixers/short_conv.py +0 -0
  75. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo/modules/token_mixers/state/__init__.py +0 -0
  76. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo/modules/token_mixers/state/common.py +0 -0
  77. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo/modules/token_mixers/state/kv_cache.py +0 -0
  78. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo/modules/token_mixers/state/mamba_state.py +0 -0
  79. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo/modules/token_mixers/state/short_conv_state.py +0 -0
  80. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo/modules/torch_interop.py +0 -0
  81. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo/modules/transformer.py +0 -0
  82. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo/modules/transformer_layer.py +0 -0
  83. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo/modules/utils.py +0 -0
  84. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo/quantization.py +0 -0
  85. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo/registry_abc.py +0 -0
  86. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo/sampling.py +0 -0
  87. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo/speculator/__init__.py +0 -0
  88. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo/speculator/common.py +0 -0
  89. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo/speculator/estimator.py +0 -0
  90. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo/speculator/inference.py +0 -0
  91. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo/speculator/ngram.py +0 -0
  92. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo/speculator/utils.py +0 -0
  93. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo/utils.py +0 -0
  94. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo.egg-info/SOURCES.txt +0 -0
  95. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo.egg-info/dependency_links.txt +0 -0
  96. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo.egg-info/entry_points.txt +0 -0
  97. {lalamo-0.5.13 → lalamo-0.5.14}/lalamo.egg-info/top_level.txt +0 -0
  98. {lalamo-0.5.13 → lalamo-0.5.14}/setup.cfg +0 -0
  99. {lalamo-0.5.13 → lalamo-0.5.14}/tests/test_cartesia_mlx_models.py +0 -0
  100. {lalamo-0.5.13 → lalamo-0.5.14}/tests/test_chat_template.py +0 -0
  101. {lalamo-0.5.13 → lalamo-0.5.14}/tests/test_generation.py +0 -0
  102. {lalamo-0.5.13 → lalamo-0.5.14}/tests/test_huggingface_model_conversion.py +0 -0
  103. {lalamo-0.5.13 → lalamo-0.5.14}/tests/test_huggingface_models.py +0 -0
  104. {lalamo-0.5.13 → lalamo-0.5.14}/tests/test_lfm2_models.py +0 -0
  105. {lalamo-0.5.13 → lalamo-0.5.14}/tests/test_mlx_models.py +0 -0
  106. {lalamo-0.5.13 → lalamo-0.5.14}/tests/test_model_spec.py +0 -0
  107. {lalamo-0.5.13 → lalamo-0.5.14}/tests/test_models.py +0 -0
  108. {lalamo-0.5.13 → lalamo-0.5.14}/tests/test_moe.py +0 -0
  109. {lalamo-0.5.13 → lalamo-0.5.14}/tests/test_parameter_tree.py +0 -0
  110. {lalamo-0.5.13 → lalamo-0.5.14}/tests/test_registry_abc.py +0 -0
{lalamo-0.5.13 → lalamo-0.5.14}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: lalamo
3
- Version: 0.5.13
3
+ Version: 0.5.14
4
4
  Summary: JAX library for optimization and export of models for use with the UZU inference engine.
5
5
  Requires-Python: <4,>=3.12
6
6
  Description-Content-Type: text/markdown
@@ -10,7 +10,7 @@ Requires-Dist: click>=8.1.8
10
10
  Requires-Dist: einops>=0.8.0
11
11
  Requires-Dist: equinox>=0.11.11
12
12
  Requires-Dist: huggingface-hub[hf-transfer]>=0.27.1
13
- Requires-Dist: jax>=0.7.2
13
+ Requires-Dist: jax>=0.8.1
14
14
  Requires-Dist: jaxtyping>=0.2.36
15
15
  Requires-Dist: jinja2>=3.1.6
16
16
  Requires-Dist: ml-dtypes>=0.5.1
@@ -23,11 +23,11 @@ Requires-Dist: safetensors>=0.6.2
23
23
  Requires-Dist: polars>=1.33.1
24
24
  Requires-Dist: xxhash>=3.5.0
25
25
  Provides-Extra: cpu
26
- Requires-Dist: jax[cpu]>=0.7.2; extra == "cpu"
26
+ Requires-Dist: jax[cpu]>=0.8.1; extra == "cpu"
27
27
  Provides-Extra: cuda
28
- Requires-Dist: jax[cuda]>=0.7.2; extra == "cuda"
28
+ Requires-Dist: jax[cuda]>=0.8.1; extra == "cuda"
29
29
  Provides-Extra: tpu
30
- Requires-Dist: jax[tpu]>=0.7.2; extra == "tpu"
30
+ Requires-Dist: jax[tpu]>=0.8.1; extra == "tpu"
31
31
  Dynamic: license-file
32
32
 
33
33
  <p align="center">
{lalamo-0.5.13 → lalamo-0.5.14}/lalamo/__init__.py RENAMED
@@ -15,7 +15,7 @@ from lalamo.speculator import (
15
15
  SpeculatorTrainingEvent,
16
16
  )
17
17
 
18
- __version__ = "0.5.13"
18
+ __version__ = "0.5.14"
19
19
 
20
20
  __all__ = [
21
21
  "AssistantMessage",
{lalamo-0.5.13 → lalamo-0.5.14}/lalamo.egg-info/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: lalamo
3
- Version: 0.5.13
3
+ Version: 0.5.14
4
4
  Summary: JAX library for optimization and export of models for use with the UZU inference engine.
5
5
  Requires-Python: <4,>=3.12
6
6
  Description-Content-Type: text/markdown
@@ -10,7 +10,7 @@ Requires-Dist: click>=8.1.8
10
10
  Requires-Dist: einops>=0.8.0
11
11
  Requires-Dist: equinox>=0.11.11
12
12
  Requires-Dist: huggingface-hub[hf-transfer]>=0.27.1
13
- Requires-Dist: jax>=0.7.2
13
+ Requires-Dist: jax>=0.8.1
14
14
  Requires-Dist: jaxtyping>=0.2.36
15
15
  Requires-Dist: jinja2>=3.1.6
16
16
  Requires-Dist: ml-dtypes>=0.5.1
@@ -23,11 +23,11 @@ Requires-Dist: safetensors>=0.6.2
23
23
  Requires-Dist: polars>=1.33.1
24
24
  Requires-Dist: xxhash>=3.5.0
25
25
  Provides-Extra: cpu
26
- Requires-Dist: jax[cpu]>=0.7.2; extra == "cpu"
26
+ Requires-Dist: jax[cpu]>=0.8.1; extra == "cpu"
27
27
  Provides-Extra: cuda
28
- Requires-Dist: jax[cuda]>=0.7.2; extra == "cuda"
28
+ Requires-Dist: jax[cuda]>=0.8.1; extra == "cuda"
29
29
  Provides-Extra: tpu
30
- Requires-Dist: jax[tpu]>=0.7.2; extra == "tpu"
30
+ Requires-Dist: jax[tpu]>=0.8.1; extra == "tpu"
31
31
  Dynamic: license-file
32
32
 
33
33
  <p align="center">
{lalamo-0.5.13 → lalamo-0.5.14}/lalamo.egg-info/requires.txt RENAMED
@@ -3,7 +3,7 @@ click>=8.1.8
3
3
  einops>=0.8.0
4
4
  equinox>=0.11.11
5
5
  huggingface-hub[hf-transfer]>=0.27.1
6
- jax>=0.7.2
6
+ jax>=0.8.1
7
7
  jaxtyping>=0.2.36
8
8
  jinja2>=3.1.6
9
9
  ml-dtypes>=0.5.1
@@ -17,10 +17,10 @@ polars>=1.33.1
17
17
  xxhash>=3.5.0
18
18
 
19
19
  [cpu]
20
- jax[cpu]>=0.7.2
20
+ jax[cpu]>=0.8.1
21
21
 
22
22
  [cuda]
23
- jax[cuda]>=0.7.2
23
+ jax[cuda]>=0.8.1
24
24
 
25
25
  [tpu]
26
- jax[tpu]>=0.7.2
26
+ jax[tpu]>=0.8.1
{lalamo-0.5.13 → lalamo-0.5.14}/pyproject.toml RENAMED
@@ -8,7 +8,7 @@ dependencies = [
8
8
  "einops>=0.8.0",
9
9
  "equinox>=0.11.11",
10
10
  "huggingface-hub[hf-transfer]>=0.27.1",
11
- "jax>=0.7.2",
11
+ "jax>=0.8.1",
12
12
  "jaxtyping>=0.2.36",
13
13
  "jinja2>=3.1.6",
14
14
  "ml-dtypes>=0.5.1",
@@ -26,9 +26,9 @@ readme = { file = "README.md", content-type = "text/markdown" }
26
26
  dynamic = ["version"]
27
27
 
28
28
  [project.optional-dependencies]
29
- cpu = ["jax[cpu]>=0.7.2"]
30
- cuda = ["jax[cuda]>=0.7.2"]
31
- tpu = ["jax[tpu]>=0.7.2"]
29
+ cpu = ["jax[cpu]>=0.8.1"]
30
+ cuda = ["jax[cuda]>=0.8.1"]
31
+ tpu = ["jax[tpu]>=0.8.1"]
32
32
 
33
33
  [dependency-groups]
34
34
  dev = [
{lalamo-0.5.13 → lalamo-0.5.14}/LICENSE RENAMED
File without changes
{lalamo-0.5.13 → lalamo-0.5.14}/README.md RENAMED
File without changes
{lalamo-0.5.13 → lalamo-0.5.14}/lalamo/common.py RENAMED
File without changes
{lalamo-0.5.13 → lalamo-0.5.14}/lalamo/data/__init__.py RENAMED
File without changes
{lalamo-0.5.13 → lalamo-0.5.14}/lalamo/data/utils.py RENAMED
File without changes
{lalamo-0.5.13 → lalamo-0.5.14}/lalamo/main.py RENAMED
File without changes
{lalamo-0.5.13 → lalamo-0.5.14}/lalamo/models/common.py RENAMED
File without changes
{lalamo-0.5.13 → lalamo-0.5.14}/lalamo/modules/mlp.py RENAMED
File without changes
{lalamo-0.5.13 → lalamo-0.5.14}/lalamo/modules/rope.py RENAMED
File without changes
{lalamo-0.5.13 → lalamo-0.5.14}/lalamo/modules/utils.py RENAMED
File without changes
{lalamo-0.5.13 → lalamo-0.5.14}/lalamo/quantization.py RENAMED
File without changes
{lalamo-0.5.13 → lalamo-0.5.14}/lalamo/registry_abc.py RENAMED
File without changes
{lalamo-0.5.13 → lalamo-0.5.14}/lalamo/sampling.py RENAMED
File without changes
{lalamo-0.5.13 → lalamo-0.5.14}/lalamo/utils.py RENAMED
File without changes
{lalamo-0.5.13 → lalamo-0.5.14}/setup.cfg RENAMED
File without changes
{lalamo-0.5.13 → lalamo-0.5.14}/tests/test_models.py RENAMED
File without changes
{lalamo-0.5.13 → lalamo-0.5.14}/tests/test_moe.py RENAMED
File without changes