Skip to main content

Differential geometry with jax

Project description

diffjeom

Differential geometry powered by jax. Given a metric function, this package lets you compute:

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


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)

Uploaded Source

Built Distribution

diffjeom-0.0.1-py3-none-any.whl (4.7 kB view details)

Uploaded Python 3

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

Hashes for diffjeom-0.0.1.tar.gz
Algorithm Hash digest
SHA256 2fb05845c43d88258474e2eb21f7527f47c69370f90d7abec89370c401970c53
MD5 4b8d30cd2db4d1b564f3f9d63f232faa
BLAKE2b-256 a6f056f83a7392efba30a8159dfe275ac21fefc995c1922892694422341298a8

See more details on using hashes here.

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

Hashes for diffjeom-0.0.1-py3-none-any.whl
Algorithm Hash digest
SHA256 3e7cd395f8107064bb63c6bd79791dfb660ece9794c81b9dd1c24c650b233d0a
MD5 52ce26c8cf1abfa2554b9c0bb8249f9f
BLAKE2b-256 62e515e7686c8741729ba6c6ca2c10dd52e0d630bc655450e9851d02f385894f

See more details on using hashes here.

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page