Tools for JAX.

Navigation

Verified details

These details have been verified by PyPI
Maintainers
Avatar for neil.g from gravatar.com neil.g

Unverified details

These details have not been verified by PyPI
Project links
Meta
  • License: MIT License (MIT)
  • Author: Neil Girdhar
  • Requires: Python >=3.8, <4.0
Classifiers

Project description

This repository implements a variety of tools for the differential programming library JAX. It includes:

  • A dataclass decorator that facilitates defining JAX trees, provides convenient text display, and provides a mypy plugin

  • A custom VJP decorator that supports both static and non-differentiable arguments

  • A random number generator class

  • JAX tree registration for NetworkX graph types

  • Testing tools that automatically produce testing code

Contribution guidelines

  • Conventions: PEP8.

  • How to run tests: pytest .

  • How to clean the source:

    • isort tjax

    • pylint tjax

    • mypy tjax

    • flake8 tjax

Project details

Verified details

These details have been verified by PyPI
Maintainers
Avatar for neil.g from gravatar.com neil.g

Unverified details

These details have not been verified by PyPI
Project links
Meta
  • License: MIT License (MIT)
  • Author: Neil Girdhar
  • Requires: Python >=3.8, <4.0
Classifiers

Release history Release notifications | RSS feed

1.1.0

1.0.9

1.0.8

1.0.7

1.0.6

1.0.5

1.0.4

1.0.3

1.0.2

1.0.1

1.0.0

0.32.6

0.32.5

0.32.4

0.32.3

0.32.2

0.32.1

0.32.0

0.31.14

0.31.13

0.31.12

0.31.11

0.31.10

0.31.9

0.31.8

0.31.7

0.31.6

0.31.5

0.31.4

0.31.3

0.31.2

0.31.1

0.31.0

0.30.1

0.30.0

0.29.7

0.29.6

0.29.5

0.29.4

0.29.3

0.29.2

0.29.1

0.29.0

0.28.4

0.28.3

0.28.2

0.28.1

0.28.0

0.27.0

0.26.0

0.25.3

0.25.2

0.25.1

0.25.0

0.24.1

0.24.0

0.23.1

0.23.0

0.22.4

0.22.3

0.22.2

0.22.1

0.22.0

0.21.3

0.21.2

0.21.1

0.21.0

0.20.1

0.20.0

0.19.8

0.19.7

0.19.6

0.19.5

0.19.4

0.19.3

0.19.2

0.19.1

0.19.0

0.18.7

0.18.6

0.18.5

0.18.4

0.18.3

0.18.2

0.18.1

0.18.0

0.17.1

0.17.0

0.16.1

0.16.0

0.15.3

0.15.2

0.15.1

0.15.0

0.14.5

0.14.4

0.14.3

0.14.2

0.14.1

0.14.0

0.13.2

0.13.1

0.13.0

0.12.6

0.12.5

0.12.4

0.12.3

0.12.2

0.12.1

0.12.0

0.11.2

0.11.1

0.11.0

0.10.5

0.10.4

0.10.3

0.10.2

0.10.1

0.10.0

0.9.0

0.8.4

0.8.3

0.8.2

0.8.1

0.8.0

0.7.13

0.7.12

0.7.11

0.7.10

0.7.9

0.7.8

0.7.7

0.7.6

0.7.5

0.7.4

0.7.3

0.7.2

0.7.1

0.7.0

0.6.9

0.6.8

0.6.7

0.6.6

0.6.5

0.6.4

0.6.3

0.6.2

0.6.1

0.6

0.5.7

0.5.6

0.5.5

0.5.3

0.5.2

0.5.1

0.5

0.4.4

0.4.3

0.4.2

0.4.0

0.3.10

0.3.9

0.3.8

0.3.7

0.3.6

0.3.5

0.3.4

0.3.3

0.3.2

0.3.1

This version

0.2.1

0.2

0.1

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

tjax-0.2.1.tar.gz (9.5 kB view hashes)

Uploaded Source

Built Distribution

tjax-0.2.1-py3-none-any.whl (11.6 kB view hashes)

Uploaded Python 3

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