Skip to main content

torchmlx

torchmlx is a pytorch-shaped compatibility layer that uses mlx on apple silicon and pytorch elsewhere.

NOTE: it is experimental and built first for educational use. If you find a bug, please create an issue on the repo.

import torchmlx as torch
from torchmlx import nn, optim

model = nn.Linear(4, 2)
optimizer = optim.AdamW(model.parameters(), lr=3e-4)

see the tinystories example and compatibility details.

Metadata

Release files for pytorchmlx 0.0.1

For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.

Source distribution (sdist)

Source distribution for pytorchmlx 0.0.1
File Size Uploaded Provenance
pytorchmlx-0.0.1.tar.gz 55.2 kB Without provenance Details

Built distribution (wheel)

Table of built distributions (wheels) for pytorchmlx 0.0.1
File Interpreter ABI Platform
pytorchmlx-0.0.1-py3-none-any.whl Python 3 none any Details

Total release size: 69.2 kB

Release files / pytorchmlx-0.0.1.tar.gz

Download URL pytorchmlx-0.0.1.tar.gz
Size 55.2 kB
Tags Source
SHA-256 checksum
How to use checksums
87cbd18c5be1254a0b41839ae8ce49e35ec85d1aead23561ac331bf12ae9e1b9
BLAKE2b-256 checksum
How to use checksums
39445ed236244a753e0b95c2dcc8a466375160eb50c657b08db31f70ea49849b
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
No
Uploaded via twine/7.0.0 CPython/3.13.3

Release files / pytorchmlx-0.0.1-py3-none-any.whl

Download URL pytorchmlx-0.0.1-py3-none-any.whl
Size 14.0 kB
Tags Python 3
SHA-256 checksum
How to use checksums
76e08d2a10f8b6a6a4db0634059b4be8533a60919498ef7064ab5995de2573da
BLAKE2b-256 checksum
How to use checksums
c1f2eef4251b8dcd9a80adf8d3b0d67371339184cccb2eaf0031dda3547f5c30
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
No
Uploaded via twine/7.0.0 CPython/3.13.3

Security for pytorchmlx 0.0.1

Provenance

Provenance describes where a file came from. On PyPI, provenance is shared via attestations, which provide a verifiable record of the build or publishing details. View details, limitations and caveats.

No provenance found

None of the files in this release were uploaded with provenance metadata.

A note on package security

PyPI is a repository for distributing packages, not a service for auditing them. While third-party security researchers and our staff do their best to detect and remove malicious packages, PyPI cannot guarantee the quality, security or suitability of any given package.

To learn more about protecting yourself from malicious actors, visit our package security documentation.

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