Hi, this package is in development, not ready yet.

It is being tested and most stuff already works.

Needs more layers and archictectures.

Here is how it works:

flowchart LR
    A[You] -->|Want convenience| B(Pytorch)
    B -->|Works slowly because adds python overhead| C[CUDA]

    D[Me] --> |Want to do math| E(Rewrite popular hf models in JAX)
    E --> |Models convert to pytorch through a simple parser and prebuilt layers| F[huggingface]

JAX is fast (because of XLA compilation), works on CUDA and TPU. This repo implements DiLoCo to use any number of heterogenous compute (kaggle, molab, colab, lightning) without direct connection (only huggingface chekpointing).

Key dates

PyPI data
Data sourced directly from PyPI's database.
  • Released:
Newer release available (0.0.2.27)

1 maintainer

PyPI data
Data sourced directly from PyPI's database.
Avatar for Berrny from gravatar.com Berrny

License expression

MIT
View SPDX License List

Requires

Python >=3.11

Provides Extra

cuda cuda12 cuda13 test

Report project as malware

Download files

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

Source Distribution

adjaxt-0.0.2.23.tar.gz (41.8 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

adjaxt-0.0.2.23-py3-none-any.whl (34.6 kB view details)

Uploaded Python 3

File details

Details for the file adjaxt-0.0.2.23.tar.gz.

File metadata

  • Download URL: adjaxt-0.0.2.23.tar.gz
  • Upload date:
  • Size: 41.8 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/7.0.0 CPython/3.14.4

File hashes

Hashes for adjaxt-0.0.2.23.tar.gz
Algorithm Hash digest
SHA256 6a9adb5e60f512c8db8d6f8180f38c338aa33c4a14bc9ff974532dee6ae9bfcd
MD5 b253c3edf8d899854966b7d73a32eaa8
BLAKE2b-256 5685945aa1020fcaeb2557d7386b1fab1f23e3fe3d7feaf57e43c19c5ffe55bc

See more details on using hashes here.

File details

Details for the file adjaxt-0.0.2.23-py3-none-any.whl.

File metadata

  • Download URL: adjaxt-0.0.2.23-py3-none-any.whl
  • Upload date:
  • Size: 34.6 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/7.0.0 CPython/3.14.4

File hashes

Hashes for adjaxt-0.0.2.23-py3-none-any.whl
Algorithm Hash digest
SHA256 d3736c31a6ec7bde3e445956d163cecc288d3da86ccf65a70965f4d0077a430a
MD5 887506144d8168abec074938312cbe01
BLAKE2b-256 d0cc3c61baf24a377cfe7b9a9c9ac3c68ee74f4f9d6e798b9143b3662f9bf765

See more details on using hashes here.

Release history Release notifications | RSS feed

0.0.2.27

2 files

0.0.2.26

2 files

0.0.2.25

2 files

0.0.2.24

2 files

This release

0.0.2.23 This release

2 files

0.0.2.22

2 files

0.0.2.21

2 files

0.0.2.20

2 files

0.0.2.19

2 files

0.0.2.18

2 files

0.0.2.17

2 files

0.0.2.16

2 files

0.0.2.15

2 files

0.0.2.14

2 files

0.0.2.13

2 files

0.0.2.12

2 files

0.0.2.11

2 files

0.0.2.10

2 files

0.0.2.9

2 files

0.0.2.8

2 files

0.0.2.7

2 files

0.0.2.6

2 files

0.0.2.5

2 files

0.0.2.4

2 files

0.0.2.3

2 files

0.0.2.2

2 files

0.0.2.1

2 files

0.0.2.0

2 files

0.0.1.5

2 files

0.0.1.4

2 files

0.0.1.3

2 files

0.0.1.2

2 files

0.0.1.1

2 files

0.0.1

2 files

Anthropic, PBC Visionary sponsor Bloomberg Visionary sponsor Hudson River Trading Visionary sponsor Meta Visionary sponsor NVIDIA Visionary sponsor Microsoft Sustainability sponsor Depot Continuous Integration AWS Cloud computing and Security Sponsor Datadog Monitoring Fastly CDN Google Download Analytics Sentry Error logging StatusPage Status page