| Name | Version | Summary | date |
| JaxSSO |
1.0.0 |
A framework for structural shape optimization based on automatic differentiation (AD) and the adjoint method, enabled by JAX |
2024-07-29 02:10:41 |
| gwkokab |
0.1.0 |
A JAX-based gravitational-wave population inference |
2024-07-26 21:07:52 |
| jaxtra |
0.0.2 |
JAXTRA |
2024-07-09 04:41:43 |
| transformers-backport |
4.33.4 |
transformers backport - version 4.33.x |
2024-07-04 23:15:53 |
| roger |
3.0.8 |
Runoff Generation Research - a process-based hydrological toolbox model in Python |
2024-07-03 10:47:23 |
| mewtax |
0.0.1 |
Differentiable minimization in jax using Newton's method. |
2024-06-14 19:52:51 |
| FJFormer |
0.0.66 |
Embark on a journey of paralleled/unparalleled computational prowess with FJFormer - an arsenal of custom Jax Flax Functions and Utils that elevate your AI endeavors to new heights! |
2024-06-12 13:32:40 |
| EasyDeL |
0.0.67 |
An open-source library to make training faster and more optimized in Jax/Flax |
2024-06-02 17:14:40 |
| bosonic |
0.1.1 |
Bosonic JAX |
2024-05-27 05:03:16 |
| nanodl |
1.2.5.dev1 |
A Jax-based library for designing and training transformer models from scratch. |
2024-05-26 21:28:50 |
| diffusersv |
0.28.0.dev0 |
State-of-the-art diffusion in PyTorch and JAX. |
2024-05-17 02:57:47 |
| clu |
0.0.12 |
Set of libraries for ML training loops in JAX. |
2024-04-10 20:36:06 |
| sobek |
0.0.0 |
Computer vision for JAX |
2024-04-03 13:41:29 |
| fenbux |
0.1.0 |
A Simple Statistical Distribution Library In JAX |
2024-03-30 07:02:44 |
| autobnn |
0.0.2.dev0 |
Package for training Gaussian process-like Bayesian Neural Networks with composite structure. |
2024-03-28 00:19:39 |
| tf2jax |
0.3.6 |
TF2JAX: Convert TensorFlow to JAX |
2024-03-26 10:38:21 |
| souJpg-diffusers |
0.27.2 |
State-of-the-art diffusion in PyTorch and JAX. |
2024-03-25 00:20:06 |
| hydrax |
0.2.2 |
A zero-copy multiprocess dataloader for JAX. |
2024-03-24 19:43:42 |
| dataclass-array |
1.5.2 |
Dataclasses that behave like numpy arrays (with indexing, slicing, vectorization). |
2024-03-19 15:25:17 |
| feedbax |
0.1.2 |
Optimal feedback control + interventions in JAX. |
2024-03-13 14:44:59 |