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 1 year ago
Linked from 1 awesome list

cudajaxxla

Backlinks from these awesome lists:

Related projects:

Repository Description Stars
jeremiecoullon/jax-tqdm Library that enhances JAX with support for dynamic progress bars in scans and loops 100
dwqs/mp-jithub A mini program for interacting with GitHub 30
appliedsec/djangular A Django app that provides tools and integration for AngularJS applications. 219
gordicaleksa/get-started-with-jax A repository providing tutorials and resources to learn JAX, a popular alternative to PyTorch and TensorFlow for machine learning. 670
nvidia/jax-toolbox Provides optimized tools and infrastructure for JAX development on NVIDIA GPUs 268
shigekikarita/tfd A D programming language wrapper around the TensorFlow C API 32
derelictorg/derelictcuda A dynamic binding to the CUDA library for use in programming languages 17
b-straub/dexienet A .NET wrapper for IndexedDB that provides a simple interface for building cloud-enabled databases 44
kattunga/blazorindexeddbjs A Blazor library for accessing IndexedDB 11
ankurp/cent Extensions for Swift Standard Types and Classes providing useful functional programming helper methods. 227
zdharma-continuum/zgdbm An extension to a GDB debugger for C code with customizable build and library settings 0
jaxgaussianprocesses/gpjax Provides a low-level interface to Gaussian process models in JAX for flexible extension and customisation 467
dlang/visuald Provides integration and tools for the D programming language in Visual Studio 288
tmcdonell/cuda FFI bindings to CUDA for Haskell programming 75
acdemiralp/gl A C++ wrapper around the OpenGL 4.6 Core Profile for creating and managing graphics resources. 158