https://github.com/amimre/spbessax

Spherical bessel functions for JAX
https://github.com/amimre/spbessax

bessel-function jax spherical-bessel

Last synced: 2 months ago
JSON representation

Spherical bessel functions for JAX

• Host: GitHub
• URL: https://github.com/amimre/spbessax
• Owner: amimre
• License: apache-2.0
• Created: 2024-08-01T08:14:15.000Z (over 1 year ago)
• Default Branch: main
• Last Pushed: 2024-11-12T13:01:28.000Z (over 1 year ago)
• Last Synced: 2024-11-12T13:42:37.494Z (over 1 year ago)
• Topics: bessel-function, jax, spherical-bessel
• Language: Python
• Homepage:
• Size: 40 KB
• Stars: 2
• Watchers: 1
• Forks: 0
• Open Issues: 1
• Metadata Files:
  • Readme: README.md
  • License: LICENSE
  • Authors: AUTHORS

Awesome Lists containing this project

README

          
# spbessax

This repository contains a JAX implementation of spherical bessel functions that

is fully compatible with jax.jit and jax.grad. It is based on the implementation

by @sousaw in [NeuralIL](https://github.com/Madsen-s-research-group/neuralil-public-releases).

The package can be installed from [PyPi](https://pypi.org/project/spbessax/) using pip:

`pip install psbessax`