extending-jax
by dfm
Extending JAX with custom C++ and CUDA code
AI summary
JAX wrapper
This project provides infrastructure to interface custom C++ and CUDA code with the JAX library for scientific computing
- stars
- 379
- forks
- 23
- watching
- 10
- awesome list
- 1
Similar projects
Found by comparing what the projects do, not just their names.
JAX progress bar library
Library that enhances JAX with support for dynamic progress bars in scans and loops
Github wrapper
A mini program for interacting with GitHub
Angular framework wrapper
A Django app that provides tools and integration for AngularJS applications.
JAX tutorial series
A repository providing tutorials and resources to learn JAX, a popular alternative to PyTorch and TensorFlow for machine learning.
JAX toolbox
Provides optimized tools and infrastructure for JAX development on NVIDIA GPUs
Tensorflow wrapper
A D programming language wrapper around the TensorFlow C API
CUDA library wrapper
A dynamic binding to the CUDA library for use in programming languages
IndexedDB wrapper
A .NET wrapper for IndexedDB that provides a simple interface for building cloud-enabled databases
IndexedDB wrapper
A Blazor library for accessing IndexedDB
ankurp/cent227
Array extender
Extensions for Swift Standard Types and Classes providing useful functional programming helper methods.
Debugger extension
An extension to a GDB debugger for C code with customizable build and library settings
GP model library
Provides a low-level interface to Gaussian process models in JAX for flexible extension and customisation
D compiler extension
Provides integration and tools for the D programming language in Visual Studio
CUDA FFI
FFI bindings to CUDA for Haskell programming
OpenGL wrapper
A C++ wrapper around the OpenGL 4.6 Core Profile for creating and managing graphics resources.
Featured in 1 awesome list
Each link jumps to the spot where the list mentions extending-jax.
