Boris Zimka

Wrap NeoOnnx into NeoML/Python

1

In most cases .onnx models are produced from Python environment - e.g. from `tensorflow + tf2onnx` or `torch + torch.onnx.export`). Currently to convert such exported onnx models into NeoML you...

Set CUDA_ACHITECUTRE to native when undefined for NeoML/Python

2

Whenever I try to build NeoML/Python with `python setup.py install`, I get the following error: ``` ... -- Configuring done CMake Error in CMakeLists.txt: CUDA_ARCHITECTURES is empty for target "PythonWrapper"....

einops.layers.torch.Rearrange does not accept a list[torch.Tensor] as an input

1

**Describe the bug** `einops.layers.torch.Rearrange` and `einops.rearrange` behave differently with `list[torch.Tensor]`. It is possible to use `rearrange` for concatenation, while `Rearrange` fails with error: > einops/_torch_specific.py:66, in TorchJitBackend.shape(x) 64 @staticmethod 65...

## What this does This PR fixes #470 (🐛 Bug) The issue is caused by recent v1.0.0 release of `gymnasium` from 08.10.2024. It looks like the `final_info` key-value pair is...

### System Info ```Shell - `lerobot` version: 0.1.0 - Platform: Linux-6.5.0-41-generic-x86_64-with-glibc2.35 - Python version: 3.10.15 - Huggingface_hub version: 0.25.2 - Dataset version: 3.0.1 - Numpy version: 1.26.4 - PyTorch version...