dmdaksh 7d7e236061 - Removed unused Python imports (#683) ... - bert/model.py:10: tree_unflatten - bert/model.py:2: dataclass - bert/model.py:8: numpy - cifar/resnet.py:6: Any - clip/model.py:15: tree_flatten - clip/model.py:9: Union - gcn/main.py:8: download_cora - gcn/main.py:9: cross_entropy - llms/gguf_llm/models.py:12: tree_flatten, tree_unflatten - llms/gguf_llm/models.py:9: numpy - llms/mixtral/mixtral.py:12: tree_map - llms/mlx_lm/models/dbrx.py:2: Dict, Union - llms/mlx_lm/tuner/trainer.py:5: partial - llms/speculative_decoding/decoder.py:1: dataclass, field - llms/speculative_decoding/decoder.py:2: Optional - llms/speculative_decoding/decoder.py:5: mlx.nn - llms/speculative_decoding/decoder.py:6: numpy - llms/speculative_decoding/main.py:2: glob - llms/speculative_decoding/main.py:3: json - llms/speculative_decoding/main.py:5: Path - llms/speculative_decoding/main.py:8: mlx.nn - llms/speculative_decoding/model.py:6: tree_unflatten - llms/speculative_decoding/model.py:7: AutoTokenizer - llms/tests/test_lora.py:13: yaml_loader - lora/lora.py:14: tree_unflatten - lora/models.py:11: numpy - lora/models.py:3: glob - speechcommands/kwt.py:1: Any - speechcommands/main.py:7: mlx.data - stable_diffusion/stable_diffusion/model_io.py:4: partial - whisper/benchmark.py:5: sys - whisper/test.py:5: subprocess - whisper/whisper/audio.py:6: Optional - whisper/whisper/decoding.py:8: mlx.nn		2024-04-16 07:50:32 -07:00
..
__init__.py	Mlx llm package (#301)	2024-01-12 10:25:56 -08:00
base.py	Mlx llm package (#301)	2024-01-12 10:25:56 -08:00
cohere.py	Fix for cohere plus (#650)	2024-04-05 14:11:24 -07:00
dbrx.py	- Removed unused Python imports (#683)	2024-04-16 07:50:32 -07:00
gemma.py	Configurable LR schedulers (#604)	2024-03-29 13:41:10 -07:00
llama.py	Switch to fast RMS/LN Norm (#603)	2024-03-23 07:13:51 -07:00
mixtral.py	Fix argpartition call in Mixtral and other MOES (#676)	2024-04-12 11:00:56 -07:00
olmo.py	Configurable LR schedulers (#604)	2024-03-29 13:41:10 -07:00
phi.py	Switch to fast RMS/LN Norm (#603)	2024-03-23 07:13:51 -07:00
phixtral.py	Fix argpartition call in Mixtral and other MOES (#676)	2024-04-12 11:00:56 -07:00
plamo.py	Configurable LR schedulers (#604)	2024-03-29 13:41:10 -07:00
qwen2_moe.py	Add support for qwen2moe (#640)	2024-04-02 11:33:29 -07:00
qwen2.py	Switch to fast RMS/LN Norm (#603)	2024-03-23 07:13:51 -07:00
qwen.py	Switch to fast RMS/LN Norm (#603)	2024-03-23 07:13:51 -07:00
stablelm.py	Stable lm 2 (#666)	2024-04-08 14:18:55 -07:00
starcoder2.py	Switch to fast RMS/LN Norm (#603)	2024-03-23 07:13:51 -07:00