Skip to content

Commit ed0e759

Browse files
authored
Update README CLAX preprint
1 parent eede292 commit ed0e759

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

README.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,9 @@
11
# CLAX: Fast and Flexible Neural Click Models in JAX
22

3-
[CLAX]() is a modular framework to build click models with gradient-based optimization in [JAX](https://github.com/jax-ml/jax) and [Flax NNX](https://flax.readthedocs.io/en/v0.8.3/experimental/nnx/index.html).
3+
[CLAX](https://arxiv.org/pdf/2511.03620) is a modular framework to build click models with gradient-based optimization in [JAX](https://github.com/jax-ml/jax) and [Flax NNX](https://flax.readthedocs.io/en/v0.8.3/experimental/nnx/index.html).
44
CLAX is built to be fast, providing orders of magnitudes speed-up compared to classic EM-based frameworks, such as [PyClick](https://github.com/markovi/PyClick), by leveraging auto-diff and vectorized computations on GPUs.
55

6-
The current documentation is available [here](https://philipphager.github.io/clax/) and our pre-print [here]().
6+
The current documentation is available [here](https://philipphager.github.io/clax/) and our pre-print [here](https://arxiv.org/pdf/2511.03620).
77

88
## Installation
99
CLAX requires JAX. For installing JAX with CUDA support, please refer to the [JAX documentation](https://github.com/jax-ml/jax?tab=readme-ov-file#installation). CLAX itself is available via pypi:

0 commit comments

Comments
 (0)