Update README.md

README.md

@@ -1,6 +1,8 @@
 # SSM: Bayesian learning and inference for state space models
 [![Test status](https://travis-ci.org/lindermanlab/ssm.svg?branch=master)](https://travis-ci.org/lindermanlab/ssm)
 
+**Note: We're working full time on a [JAX](https://github.com/google/jax) refactor of SSM that will take advantage of JIT compilation, GPU and TPU support, automatic differentation, etc. Unfortunately, we'll only be able to offer minimal support on this version of SSM. Please stay tuned for updates soon!**
+
 This package has fast and flexible code for simulating, learning, and performing inference in a variety of state space models.
 Currently, it supports: