{"repo":"sashakolpakov/dire-jax","free":true,"listed":false,"github":"https://github.com/sashakolpakov/dire-jax","clone":"git clone https://github.com/sashakolpakov/dire-jax.git","description":"DImensionality REduction in JAX","language":"Python","stars":27,"topics":["data-science","data-visualization","dimensionality-reduction","embeddings","jax","machine-learning","pca","random-projection","tsne","umap"],"license":"Apache-2.0","category":"machine-learning","readme_excerpt":"A high-performance DImensionality REduction package with JAX DiRe offers fast dimensionality reduction preserving the global dataset structure, with benchmarks showing competitive performance against UMAP and t-SNE. Built with JAX for efficient computation on CPUs and GPUs. Quick start Basic installation (JAX backend only): With utilities for benchmarking: Complete installation with utilities: Note : For GPU or TPU acceleration, JAX needs to be specifically installed with hardware support. See the JAX documentation for more details on enabling GPU/TPU support. Example usage: The output should look similar to Documentation Please refer to the DiRe API documentation for more instructions. Project documentation structure: - /docs/ - API documentation and architecture details - /benchmarking/ - Performance benchmarks and scaling results - /examples/ - Example usage and demos - /tests/ - Test suite and benchmarking notebooks Working paper Our working paper is available on the arXiv. Performance Characteristics DiRe-JAX is optimized for small-medium datasets ( 65K points Benchmarking and utilities For benchmarking utilities and quality metrics: This provides access to dimensionality reduction quality metrics and benchmarking routines. Some utilities use external packages for persistent homology computations which may increase runtime. Contributing Please follow the contibuting guide. Thanks! Citation If you use this work, please cite it as: BibTeX: APA Style: Acknowledgement This w","default_branch":null,"files":null,"tree":[],"storefront":"/r/sashakolpakov","claimed":false,"request_supported":{"post":"https://gitbuyer.com/r/sashakolpakov/dire-jax/request-supported","requests":0},"note":"indexed from public GitHub; nothing is for sale on this page. Clone it from GitHub. Paid listings live at /search."}