https://repo.fury.io/emaballarin/
| Name | Kind | Latest version | |
|---|---|---|---|
| accelerate | python | 1.12.0.dev0 | |
| Accelerate | |||
| advertorch | python | 0.2.4 | |
| No description | |||
| clu | python | 0.0.12 | |
| Set of libraries for ML training loops in JAX. | |||
| diffrax | python | 0.7.0 | |
| GPU+autodiff-capable ODE/SDE/CDE solvers written in JAX. | |||
| einops | python | 0.8.2 | |
| A new flavour of deep learning operations | |||
| einshape | python | 1.0 | |
| DSL-based reshaping library for JAX and other frameworks | |||
| equinox | python | 0.13.2 | |
| Elegant easy-to-use neural networks in JAX. | |||
| gin-config | python | 0.5.0 | |
| Gin-Config: A lightweight configuration library for Python | |||
| haliax | python | 1.4 | |
| Named Tensors for Legible Deep Learning in JAX | |||
| hydra-core | python | 1.4.0.dev1 | |
| A framework for elegantly configuring complex applications | |||
| jaxtyping | python | 0.3.3 | |
| Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. | |||
| jmp | python | 0.0.6 | |
| JMP is a Mixed Precision library for JAX. Forked from DeepMind's JMP. | |||
| lineax | python | 0.0.8 | |
| Linear solvers in JAX and Equinox. | |||
| lsuv | python | 0.2.2 | |
| LSUV initialization for PyTorch | |||
| ml_collections | python | 1.1.0 | |
| ML Collections is a library of Python collections designed for ML usecases. | |||
| ml_dtypes | python | 0.5.3 | |
| ml_dtypes is a stand-alone implementation of several NumPy dtype extensions used in machine learning. | |||
| neuraloperator | python | 2.0.0 | |
| NeuralOperator: Learning in Infinite Dimensions | |||
| omegaconf | python | 2.4.0.dev3 | |
| A flexible configuration library | |||
| open3d | python | 0.19.0 | |
| Open3D: A Modern Library for 3D Data Processing. | |||
| open3d-cpu | python | 0.19.0 | |
| Open3D: A Modern Library for 3D Data Processing. | |||