6 projects
smcx
State-space inference in JAX: Kalman and particle filters, tempered SMC, and SMC2 across CPU and accelerators
pypkgkit
A template for python packages
lmxlab
Transformer language models on Apple Silicon with MLX
smcjax
Sequential Monte Carlo and particle filtering in JAX
pylmt
A PyTorch implementation of transformer-based language models including GPT architecture for pretraining and fine-tuning
