JAX-Toolbox
JAX GPU toolbox
A collection of optimized JAX libraries and examples for simplified development on NVIDIA GPUs
JAX-Toolbox
245 stars
22 watching
48 forks
Language: Jupyter Notebook
last commit: 6 days ago
Linked from 1 awesome list
n2cholas/awesome-jaxRelated projects:
| Repository | Description | Stars |
|---|---|---|
 alecjacobson/gptoolbox | A collection of utility functions for geometry processing and related tasks in MATLAB. | 629 |
 eclipse-openj9/openj9-utils | An interim collection of reusable tools and utilities for developing and experimenting with the OpenJ9 project. | 16 |
 jidesoft/jidefx-oss | A JavaFX toolkit containing various utilities and extensions | 106 |
 run-ai/jupyterlab_genv | A JupyterLab extension for managing GPU resources. | 20 |
 gpustack/gpustack | Manages GPU clusters for large language models | 600 |
 gpeyre/matlab-toolboxes | A collection of mature toolboxes for Matlab that provide specialized functionality. | 323 |
 jaywcjlove/jaywcjlove | A collection of tutorials, articles, and tools for learning web development, including macOS/iOS app development, Git, and GitHub Actions. | 40 |
 mapbox/mapboxgl-jupyter | A library that enables native data visualization in Jupyter Notebooks using WebGL-based Mapbox GL JS | 668 |
| nvidia/multi-gpu-programming-models | A collection of examples demonstrating various approaches to programming multiple GPUs in parallel | 557 |
| nvidia/matx | A C++17 GPU-accelerated numerical computing library with Python-like syntax | 1,220 |
 studiohyperdrive/ngx-tools | A collection of libraries providing tooling and utilities for Angular projects | 10 |
 joncox123/cortexsys | A toolbox for building and training deep neural networks in Matlab | 70 |
 isaacgerg/matlabhyperspectraltoolbox | A MATLAB toolbox providing algorithms and functions for hyperspectral image processing and analysis | 106 |
| nvidia/grcuda | Enables efficient data exchange and invocation of existing GPU kernels from host languages in the GraalVM | 222 |
 gentilkiwi/kekeo | A toolbox to manipulate Microsoft Kerberos in C. | 1,425 |