extending-jax

JAX wrapper

This project provides infrastructure to interface custom C++ and CUDA code with the JAX library for scientific computing

Extending JAX with custom C++ and CUDA code

GitHub

379 stars
10 watching
23 forks
Language: Python
last commit: about 2 years ago
Linked from 1 awesome list

cudajaxxla

Backlinks from these awesome lists:

Related projects:

RepositoryDescriptionStars
jeremiecoullon/jax-tqdmLibrary that enhances JAX with support for dynamic progress bars in scans and loops100
dwqs/mp-jithubA mini program for interacting with GitHub30
appliedsec/djangularA Django app that provides tools and integration for AngularJS applications.219
gordicaleksa/get-started-with-jaxA repository providing tutorials and resources to learn JAX, a popular alternative to PyTorch and TensorFlow for machine learning.670
nvidia/jax-toolboxProvides optimized tools and infrastructure for JAX development on NVIDIA GPUs268
shigekikarita/tfdA D programming language wrapper around the TensorFlow C API32
derelictorg/derelictcudaA dynamic binding to the CUDA library for use in programming languages17
b-straub/dexienetA .NET wrapper for IndexedDB that provides a simple interface for building cloud-enabled databases44
kattunga/blazorindexeddbjsA Blazor library for accessing IndexedDB11
ankurp/centExtensions for Swift Standard Types and Classes providing useful functional programming helper methods.227
zdharma-continuum/zgdbmAn extension to a GDB debugger for C code with customizable build and library settings0
jaxgaussianprocesses/gpjaxProvides a low-level interface to Gaussian process models in JAX for flexible extension and customisation467
dlang/visualdProvides integration and tools for the D programming language in Visual Studio288
tmcdonell/cudaFFI bindings to CUDA for Haskell programming75
acdemiralp/glA C++ wrapper around the OpenGL 4.6 Core Profile for creating and managing graphics resources.158