JAX-Toolbox

JAX toolbox

Provides optimized tools and infrastructure for JAX development on NVIDIA GPUs

JAX-Toolbox

GitHub
268 stars
21 watching
50 forks
Language: Jupyter Notebook
last commit: about 5 hours ago
Linked from 1 awesome list


Backlinks from these awesome lists:

Related projects:

Repository Description Stars
alecjacobson/gptoolbox A collection of utility functions for geometry processing and related tasks in MATLAB. 632
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 107
run-ai/jupyterlab_genv A JupyterLab extension for managing GPU resources. 20
gpustack/gpustack Manages GPU clusters for running AI models 850
gpeyre/matlab-toolboxes A collection of mature toolboxes for Matlab that provide specialized functionality. 326
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 670
nvidia/multi-gpu-programming-models A collection of examples demonstrating various approaches to programming multiple GPUs in parallel 575
nvidia/matx A C++17 GPU-accelerated numerical computing library with Python-like syntax 1,229
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 223
gentilkiwi/kekeo A toolbox to manipulate Microsoft Kerberos in C. 1,436