Simple Linear Neuron Boosting in JAX

Paper

This implementation is compatible with Optax

Example Usage

uv sync
uv run jupyter notebook examples/mnist.ipynb

Citation

@misc{munoz2025simplelinearneuronboosting,
      title={Simple Linear Neuron Boosting}, 
      author={Daniel Munoz},
      year={2025},
      eprint={2502.01131},
      archivePrefix={arXiv},
      primaryClass={cs.LG},
      url={https://arxiv.org/abs/2502.01131}, 
}