Skip to content

Commit 89c6ee3

Browse files
committed
add acknowledgment
1 parent 19fd99f commit 89c6ee3

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

README.md

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -176,6 +176,14 @@ print("GPU support:", hasattr(sooki, 'gpu_ops'))
176176

177177
This project demonstrates JAX extension techniques and is provided for educational purposes.
178178

179+
## Acknowledgments
180+
181+
This project was inspired by and builds upon the excellent tutorial and examples from:
182+
183+
**[dfm/extending-jax](https://github.com/dfm/extending-jax)** - A comprehensive guide to extending JAX with custom operations
184+
185+
The original repository by Dan Foreman-Mackey provides foundational examples and best practices for JAX extensions that were instrumental in developing this project.
186+
179187
## References
180188

181189
- [JAX Custom Operations Guide](https://jax.readthedocs.io/en/latest/Custom_Operation_for_GPUs.html)

0 commit comments

Comments
 (0)