5 projects
lmxlab
Transformer language models on Apple Silicon with MLX
smcjax
Sequential Monte Carlo and particle filtering in JAX
pypkgkit
A template for python packages
pylmt
A PyTorch implementation of transformer-based language models including GPT architecture for pretraining and fine-tuning