extending-jax

Extending JAX with custom C++ and CUDA code

MIT License

Stars
368
View Code on GitHub View on X
Ecosystems: Python, Cuda

Issue Statistics

Past Year

All Time

Total Pull Requests
2
6
Merged Pull Requests
2
5
Total Issues
1
6
Time to Close Issues
about 3 hours
2 months
Badges
Extracted from project README
Tests Open In Colab
Related Projects
librapid

A highly optimised C++ library for mathematical applications and neural networks.

25 May 2021 163
https://github.com/PennyLaneAI/pennylane-lightning

The PennyLane-Lightning plugin provides a fast state-vector simulator written in C++ for use with...

06 Jul 2020 82
https://github.com/open-atmos/PySDM

Pythonic particle-based (super-droplet) warm-rain/aqueous-chemistry cloud microphysics package wi...

26 Jul 2019 45
cuda-design-patterns

Some CUDA design patterns and a bit of template magic for CUDA

16 Nov 2018 145
cccl

CUDA C++ Core Libraries

17 Sep 2020 743
nvidia-gpu-ml-library-test

Simple tests for JAX, PyTorch, and TensorFlow to test if the installed NVIDIA drivers are being p...

04 Jan 2021 15
https://github.com/mp3guy/Kintinuous

Real-time large scale dense visual SLAM system

22 Oct 2015 913
cxbqn

BQN virtual machine

22 Oct 2021 29
spbla

Sparse Boolean linear algebra for Nvidia Cuda, OpenCL and CPU computations

18 Feb 2021 14
awesome-gpgpu

A curated list of awesome GPGPU (CUDA/OpenCL/Vulkan) resources

20 Jun 2018 63
https://github.com/mp3guy/ElasticFusion

Real-time dense visual SLAM system

22 Oct 2015 1,772
RadonKA.jl

A simple yet sufficiently fast (attenuated) Radon and backproject implementation using KernelAbst...

20 Jun 2023 7
libCEED

CEED Library: Code for Efficient Extensible Discretizations

06 Dec 2017 179
awesome-rust-list

This repository lists some awesome public Rust projects, Videos, Blogs and Jobs.

07 Aug 2022 34
https://github.com/MrNeRF/gaussian-splatting-cuda

3D Gaussian Splatting, reimagined: Unleashing unmatched speed with C++ and CUDA from the ground up!

30 Jul 2023 862