Differential geometry with jax
Project description
diffjeom
Differential geometry powered by jax. Given a metric function, this package lets you compute:
- Christoffel symbols of the second kind (
get_christoffel2
); - The Riemann curvature tensor (
get_riemann
); - The Ricci tensor (
get_ricci_tensor
); - The Ricci scalar/scalar curvature (
get_ricci_scalar
).
These work with the usual jax transformations like jit
and vmap
.
The functions check_christoffel2_sym
, check_riemann_sym
,
check_diff_bianchi
and check_ricci_tensor_sym
are provided to check
identities for these objects.
Getting started
Install with
pip install diffjeom
This will install jax and jaxlib if you don't already have them. The tests
require numpy and can be run with pytest
. Check them out for some usage
examples.
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
diffjeom-0.0.1.tar.gz
(2.9 kB
view details)
Built Distribution
File details
Details for the file diffjeom-0.0.1.tar.gz
.
File metadata
- Download URL: diffjeom-0.0.1.tar.gz
- Upload date:
- Size: 2.9 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/3.4.1 importlib_metadata/3.7.3 pkginfo/1.7.0 requests/2.25.1 requests-toolbelt/0.9.1 tqdm/4.58.0 CPython/3.9.1
File hashes
Algorithm | Hash digest | |
---|---|---|
SHA256 | 2fb05845c43d88258474e2eb21f7527f47c69370f90d7abec89370c401970c53 |
|
MD5 | 4b8d30cd2db4d1b564f3f9d63f232faa |
|
BLAKE2b-256 | a6f056f83a7392efba30a8159dfe275ac21fefc995c1922892694422341298a8 |
File details
Details for the file diffjeom-0.0.1-py3-none-any.whl
.
File metadata
- Download URL: diffjeom-0.0.1-py3-none-any.whl
- Upload date:
- Size: 4.7 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/3.4.1 importlib_metadata/3.7.3 pkginfo/1.7.0 requests/2.25.1 requests-toolbelt/0.9.1 tqdm/4.58.0 CPython/3.9.1
File hashes
Algorithm | Hash digest | |
---|---|---|
SHA256 | 3e7cd395f8107064bb63c6bd79791dfb660ece9794c81b9dd1c24c650b233d0a |
|
MD5 | 52ce26c8cf1abfa2554b9c0bb8249f9f |
|
BLAKE2b-256 | 62e515e7686c8741729ba6c6ca2c10dd52e0d630bc655450e9851d02f385894f |