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
378 stars
10 watching
22 forks
Language: Python
last commit: 3 months 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 96
dwqs/mp-jithub A mini program for interacting with GitHub 30
appliedsec/djangular A Django app that provides tools and integration for AngularJS applications. 220
gordicaleksa/get-started-with-jax A repository providing tutorials and resources to learn JAX, a popular alternative to PyTorch and TensorFlow for machine learning. 661
nvidia/jax-toolbox A collection of optimized JAX libraries and examples for simplified development on NVIDIA GPUs 245
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 43
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 461
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