survival_jax A library of survival analysis models implementations done with jax for learning purposes. 2023-05: Cox Proportional Hazards